nn.AdaptiveAvgPool2d()

自适应平均池化
作用：指定输出的 H 和 W
代码：

input = torch.randn((1, 64, 8, 10))

m = nn.AdaptiveAvgPool2d((1, 1))
m1 = nn.AdaptiveAvgPool2d((None, 2))
m2 = nn.AdaptiveAvgPool2d((5, 4))

print(m(input).shape)
print(m1(input).shape)
print(m2(input).shape)

torch.Size([1, 64, 1, 1])
torch.Size([1, 64, 8, 2])
torch.Size([1, 64, 5, 4])