Pytorch 卷积过程中结果不为整数的处理

在卷积操作过程中,矩阵经卷积操作后的尺寸由以下几个因数决定:

  • 输入图片大小 W×W
  • Filter大小 F×F
  • 步长 S
  • padding的像素数 P

 经卷积后的矩阵尺寸大小计算公式为:

      N = (W − F + 2P ) / S + 1

但在实际应用中,有时会出现N为非整数的情况。

例如输入的矩阵 H=W=5,卷积核的F=2,S=2,Padding=1。

经计算我们得到的N =(5 - 2 + 2*1)/ 2 +1 = 3.5 ,此时在Pytorch中是如何处理呢?

在卷积过程中会直接将最后一行以及最后一列给忽略掉,以保证N为整数,此时N = (5 - 2 + 2*1 - 1)/ 2 + 1 = 3
 

发布了163 篇原创文章 · 获赞 14 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/qq_24502469/article/details/104173906