1. 程式人生 > >Pytorch-nn.ConvTransposed2d()

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