如何反转AvgPool2d?

时间:2018-07-28 19:28:01

标签: neural-network computer-vision conv-neural-network pytorch

是否可以在PyTorch中反转avgPool2d操作,例如将maxunpool2d转换为maxpool2d操作,如果可以,该怎么办?

我已经检查过文档,并且没有返回索引的选项,例如在maxpool2d操作中,所以我认为无法以类似的方式进行分池。

编辑: 我在Intel处找到了一个文档,其中描述了拆箱的工作方式。在检查了有关avgpool2d函数的数学运算后,解池似乎非常简单,基本上将每个输入元素镜像到多个输出元素,并应用填充以获取正确的输出大小。

1 个答案:

答案 0 :(得分:2)

我认为您正在寻找ConvTransposed2d,又名反卷积:此功能可让您“上采样”池化的层。
使用固定权重,您可以复制平均池值。您也可以训练这一层,希望会有所改善。