nn.AdaptiveAvgPool2d() 與 nn.AvgPool2d() 模塊的區別

兩者都是做二維的平均池化,但是它們關注的參數卻不一樣。 1、nn.AvgPool2d() 模塊 模塊原型爲 其中參數爲 一般我們使用它的時候,只需要關注 kernel_size 、stride 與 padding 三個參數就行了,最後輸出的尺寸爲: 其中,x 表示輸入的維度大小, y 表示對應輸出的維度大小。    2、nn.AdaptiveAvgPool2d() 模塊 相比 nn.AvgPool
相關文章
相關標籤/搜索