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()的主要内容,如果未能解决你的问题,请参考以下文章