python - 什么是自适应平均池以及它是如何工作的？

Question

我最近在尝试实现 AlexNet 时在 Pytorch 中遇到了一个方法。我不明白它是如何工作的。请用一些例子解释其背后的想法。就神经网络功能而言，它与 Maxpooling 或 Average poling 有何不同

nn.AdaptiveAvgPool2d((6, 6))

Answer 1

在平均池化或最大池化中，您基本上可以自己设置步幅和内核大小，并将它们设置为超参数。如果您碰巧更改了输入大小，则必须重新配置它们。

另一方面，在自适应池中，我们指定输出大小。步幅和内核大小会自动选择以适应需要。下面的公式用于计算源代码中的值。

Stride = (input_size//output_size)  
Kernel size = input_size - (output_size-1)*stride  
Padding = 0