完整版
H o u t = ⌊ H i n + 2 × padding [ 0 ] − dilation [ 0 ] × ( kernel_size [ 0 ] − 1 ) − 1 stride [ 0 ] + 1 ⌋ H_{out} = \left\lfloor\frac{H_{in} + 2 \times \text{padding}[0] - \text{dilation}[0] \times (\text{kernel\_size}[0] - 1) - 1}{\text{stride}[0]} + 1\right\rfloor Hout=⌊stride[0]Hin+2×padding[0]−dilation[0]×(kernel_size[0]−1)−1+1⌋
W o u t = ⌊ W i n + 2 × padding [ 1 ] − dilation [ 1 ] × ( kernel_size [ 1 ] − 1 ) − 1 stride [ 1 ] + 1 ⌋ W_{out} = \left\lfloor\frac{W_{in} + 2 \times \text{padding}[1] - \text{dilation}[1] \times (\text{kernel\_size}[1] - 1) - 1}{\text{stride}[1]} + 1\right\rfloor Wout=⌊stride[1]Win+2×padding[1]−dilation[1]×(kernel_size[1]−1)−1+1⌋
[ dilation [ 1 ] × ( kernel_size [ 1 ] − 1 ) − 1 ] [\text{dilation}[1] \times (\text{kernel\_size}[1] - 1) - 1] [dilation[1]×(kernel_size[1]−1)−1] 是加了dilation之后的kernel size。
简易版
H o u t = ⌊ H i n + 2 × padding [ 0 ] − kernel_size [ 0 ] stride [ 0 ] + 1 ⌋ H_{out} = \left\lfloor\frac{H_{in} + 2 \times \text{padding}[0] - \text{kernel\_size}[0]}{\text{stride}[0]} + 1\right\rfloor Hout=⌊stride[0]Hin+2×padding[0]−kernel_size[0]+1⌋
W o u t = ⌊ W i n + 2 × padding [ 1 ] − kernel_size [ 1 ] stride [ 1 ] + 1 ⌋ W_{out} = \left\lfloor\frac{W_{in} + 2 \times \text{padding}[1] - \text{kernel\_size}[1] }{\text{stride}[1]} + 1\right\rfloor Wout=⌊stride[1]Win+2×padding[1]−kernel_size[1]+1⌋