当前位置:网站首页>pytorch nn.AdaptiveAvgPool2d(1)

pytorch nn.AdaptiveAvgPool2d(1)

2022-07-01 03:15:00 是七叔呀

在pytorch的PSPNet中
在这里插入图片描述
网络构建有用到nn.AdaptiveAvgPool2d(pool_size)
在这里插入图片描述
其中参数pool_size为给定输出特征图的大小,其中通道数前后不发生变化。

  • 输入 ( b , c , h , w ) \color{red}{(b,c,h,w)} bchw
  • 输出 ( b , c , p o o l s i z e , p o o l s i z e ) \color{red}{(b,c,pool_size,pool_size)} bcpoolsizepoolsize

全局信息采用如下公式:
在这里插入图片描述

原网站

版权声明
本文为[是七叔呀]所创,转载请带上原文链接,感谢
https://blog.csdn.net/m0_46378271/article/details/125074085