1、m=nn.AdaptiveAvgPool2d(1)
只需要给定输出特征图的大小就好,其中通道数前后不发生变化。
全局信息采用如下公式:
采用如下的公式:
self.avgpool = nn.AdaptiveAvgPool2d(1)
输出尺寸:
torch.Size([2, 32, 16, 16])----->torch.Size([2, 32, 1, 1])
2、对于m = nn.AdaptiveMaxPool2d((None, 7))
>>> # target output size of 10x7
m = nn.AdaptiveMaxPool2d((None, 7))
input = torch.randn(1, 64, 10, 9)
output = m(input)
print(output.size())
#输出:
torch.Size([1, 64, 10, 7])