关于算子mindspore.nn.Conv2dTranspose没有output_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_channels, out_channels, kernel_size, stride=1, pad_mode="same", padding=0, dilation=1, group=1, has_bias=False, weight_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")

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值