pytorch nn.AdaptiveAvgPool2d(1)

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