自适应平均池化
作用:指定输出的 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])