adaptive_average_pool-2d code example

Example 1: adaptive_average_pool-2d

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

Example 2: adaptive_average_pool-2d

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

Example 3: adaptive_average_pool-2d

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

Example 4: adaptive_average_pool-2d

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

print(inp .shape)
print(inp)