当前位置:网站首页>自定义卷积核的分组转置卷积如何实现?

自定义卷积核的分组转置卷积如何实现?

2022-08-11 08:52:00 小乐快乐

mindspore.ops

1、在模型实现中遇到使用 torch.nn.functional.conv_transpose2d 算子,需要用到自定义卷积核,并且需要分组卷积。

2、在mindspore.ops.Conv2D存在对自定义卷积核进行卷积。

3、需要使用该算子实现从(batchsize, 12, 72,72)分三组进行转置卷积成(batchsize, 1, 144, 144),其中的stride为2,padding为0。

4、分组卷积已实现,是否有转置卷积的算子?若没有,该如何实现?

可使用:mindspore.nn.Conv2dTranspose — MindSpore master documentation

原网站

版权声明
本文为[小乐快乐]所创,转载请带上原文链接,感谢
https://blog.csdn.net/weixin_45666880/article/details/126270295