ConvTranspose2Dの出力

pytorchにおいてConv2DとConvTranspose2Dの出力の高さの計算式をここにまとめておく。

Conv2Dについては色々な文献で見つかるが、ConvTranspose2Dの出力について書かれている文献が少ないように感じる。

Conv2D

OH＝((H+2P-FH)/S) + 1

H: 入力の高さ

P: padding

FH: フィルタの高さ

S: stride

(参考：

https://blog.shikoan.com/pytorch-convtranspose2d/

)

ConvTranspose2D

OH＝(H-1)*S - 2P + FH + OP

(dialationについてははしょる)

OP: output padding

※圧縮の際はOPは使わない