卷积运算的输入输出shape

摘要

记录一下二维卷积神经网络中卷积运算的输入输出shape大小

正文

• 输入：$(H,W)$

• 卷积核：$(FH,FW)$

• 填充：$P$

• 步幅：$S$

• 输出：$(OH,OW)$

$$OH = \frac{H + 2P - FH}{S} + 1 \\ \\ OW = \frac{W + 2P - FW}{S} + 1$$