adaptive_average_pool-2d
torch.Size([1, 1, 3, 3])
tensor([[[[1., 2., 3.],
[4., 5., 6.],
[7., 8., 9.]]]])
Panicky Pigeon
torch.Size([1, 1, 3, 3])
tensor([[[[1., 2., 3.],
[4., 5., 6.],
[7., 8., 9.]]]])
tensor([[[[3., 4.],
[6., 7.]]]])
inp = torch.tensor([[[[1,2.,3], [4,5,6], [7,8,9]]]], dtype = torch.float)
print(inp .shape)
print(inp)
out = nn.AdaptiveAvgPool2d((2,2))(inp)
print(out)