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])
————————————————
版权声明:本文为CSDN博主「zouxiaolv」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/zouxiaolv/article/details/106364351

他的通道数不变,但是尺寸大小变了。变成1了。

你可能感兴趣的:(深度学习,pytorch,神经网络)