以池化层中MaxPool2d为例,Pytorch官方文档所给出内容如下:
- I n p u t : ( N , C , H i n , W i n ) Input:(N,C,H_{in},W_{in}) Input:(N,C,Hin,Win)
- O u r p u t : ( N , C , H o u t , W o u t ) Ourput:(N,C,H_{out},W_{out}) Ourput:(N,C,Hout,Wout)
其中 H o u t H_{out} Hout和 W o u t W_{out} Wout则通过下列公式计算所得:
H o u t = [ H i n + 2 × p a d d i n g [ 0 ] − d i l a t i o n [ 0 ] × ( k e r n e l _ s i z e [ 0 ] − 1 ) − 1 s t r i d e [ 0 ] + 1 ] W o u t = [ W i n + 2 × p a d d i n g [ 1 ] − d i l a t i o n [ 1 ] × ( k e r n e l _ s i z e [ 1 ] − 1 ) − 1 s t r i d e [ 1 ] + 1 ] H_{out}=[\frac{H_{in}+2\times padding[0]-dilation[0]\times (kernel\_size[0]-1)-1}{stride[0]}+1] \\ W_{out}=[\frac{W_{in}+2\times padding[1]-dilation[1]\times (kernel\_size[1]-1)-1}{stride[1]}+1] Hout=[st</