how do i use a linear layer after the AdaptiveAvgPool2d layer?
Consider this example:
m = nn.AdaptiveAvgPool2d((5,7))
dense = nn.Linear(5, 10)
input = torch.randn(1, 64, 8, 9)
output = m2(input) # its shape should be (1, 64, 5, 7)
output2 = dense(output)  # this line throws an error.
Please help me with this. Thanks in advance !