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

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章