Pytorch-nn.ConvTransposed2d()
Posted leebxo
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Pytorch-nn.ConvTransposed2d()相关的知识,希望对你有一定的参考价值。
ConvTransposed2d()其实是Conv2d()的逆过程,其参数是一样的
Conv2d():
output = (input+2*Padding-kernelSize) / stride + 1(暂时不考虑outputPadding 注意:outputPadding只是在一边Padding)
=>input = (output-1) * stride - 2*Padding + kernelSize
例如输入图片尺寸为128,inputPadding为0,kernelSize为4,stride为2,outputPadding为1,那么输出图片尺寸为64
同理可得
ConvTransposed2d():
output = (input-1) * stride - 2*Padding + kernelSize
以上是关于Pytorch-nn.ConvTransposed2d()的主要内容,如果未能解决你的问题,请参考以下文章