torch.nn.AdaptiveAvgPool2d()自适应平均池化函数

m = nn.AdaptiveAvgPool2d((5,1))
input = torch.randn(1, 64, 8, 9)
output = m(input)
print(output.shape)

torch.Size([1, 64, 5, 1])

m = nn.AdaptiveAvgPool2d((None,1))
input = torch.randn(1, 64, 8, 9)
output = m(input)
print(output.shape)

torch.Size([1, 64, 8, 1])