使用MindSpore进行ops操作(reshape, cast)在静态图中报错

问题描述:

运行环境:

Ascend910

mindspore-1.3.0

问题描述:

模型中使用ops.reshape()方法,在动态图中能正常运行静态图中报错如下:

 

RuntimeError: mindspore/ccsrc/backend/kernel_compiler/kernel_query.cc:95 KernelQueryAll] Can not find any available operator info for op [BinaryCrossEntropy, Default/network-SoftMaskedBertCLS/loss-BCELoss/BinaryCrossEntropy-op1791]. Node DebugString:kernel_graph_2:equivloss{[0]: ValueNode<Primitive> BinaryCrossEntropy, [1]: equivactive_probs, [2]: equivactive_labels, [3]: equivweight}, maybe the operator can not supported on current platform.

 trace

解答:

使用reshape之前,有没有先初始化一下: reshape = ops.Reshape()

然后再调用: output = reshape(input_x, (3, 2))

mindspore.ops.Reshape — MindSpore master documentation

另外也可以单独测一下reshape算子,看看是不是这个算子的问题

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值