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

 

你可能感兴趣的:(pytorch)