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])
本文详细介绍了PyTorch中的自适应平均池化(AdaptiveAvgPool2d)功能,展示了如何通过指定输出尺寸(H,W)进行自适应平均池化操作,并提供了具体的代码实例,演示了不同输入和输出尺寸下自适应平均池化的应用。

2万+

被折叠的 条评论
为什么被折叠?



