adaptive_average_pool-2d

torch.Size([1, 1, 3, 3])
tensor([[[[1., 2., 3.],
          [4., 5., 6.],
          [7., 8., 9.]]]])

inp = torch.tensor([[[[1,2.,3], [4,5,6], [7,8,9]]]], dtype = torch.float)

print(inp .shape)
print(inp)

tensor([[[[3., 4.],
          [6., 7.]]]])

out = nn.AdaptiveAvgPool2d((2,2))(inp)
print(out)