当前位置:   article > 正文

关于算子mindspore.nn.Conv2dTranspose没有output_padding参数_nn.convtranspose2d out_padding padding对输出特征的影响

nn.convtranspose2d out_padding padding对输出特征的影响

pytorch中的反置卷积torch.nn.ConvTranspose2d(in_channels, out_channels, kernel_size, stride=1, padding=0, output_padding=0, groups=1, bias=True, dilation=1, padding_mode='zeros')

对应mindspore是

mindspore.nn.Conv2dTranspose(in_channelsout_channelskernel_sizestride=1pad_mode="same"padding=0dilation=1group=1has_bias=Falseweight_init="normal"bias_init="zeros")

其中pytorch算子中有一个output_padding参数表示对输出的特征图的某一边进行padding,

然而mindspore算子中没有,那么如何实现对特征图的某一边进行padding呢,有替代的方案吗

pytorch中的nn.ConvTranspose2d(in_channels=num_inchannels[i],out_channels=num_inchannels[i],kernel_size=3,    stride=2, padding=1,output_padding=1,bias=False),    可以用mindspore的以下方案实现: nn.Conv2dTranspose(in_channels=num_inchannels[i],  out_channels=num_inchannels[i],  kernel_size=3,  stride=2,  padding=1,  pad_mode="pad",  has_bias=False),  nn.Pad(paddings=((0,0),(0,0),(0,1),(0,1)), mode="CONSTANT"),

本文内容由网友自发贡献,转载请注明出处:https://www.wpsshop.cn/w/你好赵伟/article/detail/154184
推荐阅读
  

闽ICP备14008679号