关于mindspore.ops.operation.Conv2d算子使用
生活随笔
收集整理的这篇文章主要介绍了
关于mindspore.ops.operation.Conv2d算子使用
小编觉得挺不错的,现在分享给大家,帮大家做个参考.
卷积层的实现算子是
mindspore.ops.operation.Conv2d(
out_channel,
kernel_size, mode=1,
pad_mode="valid",
pad=0, stride=1,
dilation=1,
group=1,
data_format="NCHW"
) 其中的mode参数用于指定不同的卷积计算方式:
mode=0,math convolution
mode=1,cross-correlation convolution
mode=2,deconvolution
mode=3,depthwise convolution
问题1:指定mode=3时报错,说mode必须为1。
问题2:将限制代码注释掉,mode设置为3,但是卷积结果和mode为1时,完全一样。这里测试时使用了相同的输入数据,使用了相同的weight。 请教正式用法,目标是实现深度可分离卷积,谢谢!
目前Conv2D算子,只支持mode=1的卷积模式,其余模式不支持。文档API已经在master已经做过修改。
详细信息可见:
mindspore.ops.Conv2D — MindSpore master 文档
总结
以上是生活随笔为你收集整理的关于mindspore.ops.operation.Conv2d算子使用的全部内容,希望文章能够帮你解决所遇到的问题。
- 上一篇: 全球及中国柔性AMOLED面板行业消费量
- 下一篇: idenet 学习记录:bili