MindSpore的nonzero算子的使用问题

在Mindspore1.8版本中,使用nonzero算子可能会遇到动态shape支持的问题,导致运行时报错。当ix是nonzero运算的结果时,会出现错误,而预定义的ix则不会。这可能是因为Mindspore对动态shape的泛化支持尚不完善。为解决此问题,建议检查网络脚本,避免使用动态shape的功能,或者等待Mindspore后续更新对相关特性的支持。目前,用户需要修改网络脚本来规避这个问题。
摘要由CSDN通过智能技术生成

问题描述:

对于1.8新增的nonzero算子, 我在使用过程中发现如下问题:

 

接着就会报错:

 

但如果ix 不是nonzero的结果,而是定义好的,就没有, 比如说:

 

希望能解决一下为什么会出现这种情况呢?

***************************解答***************************

当前mindspore 对 动态shape的支持还在完善中,可能当前泛化性没有那么好,需要查看下网络中的相关操作是否全面支持,或者查看下能否不适用动态shape的能力进行一定规避。给您造成的困扰深感歉意,目前相关需求也反应给了相应同事。当前可能只能根据进行对网络脚本进行更改规避该问题。可以后续关注mindspore相关特性的支持情况再对网络进行相关调整。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值