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