nn.AdaptiveAvgPool2d

官网

nn.AdaptiveAvgPool2d

nn.AdaptiveAvgPool2d就是自适应平均池化，指定输出（H，W）

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])