使用MindSpore中的SoftMax()算子计算单一数据出错Run op inputs type is invalid!
收藏回复举报
使用MindSpore中的SoftMax()算子计算单一数据出错Run op inputs type is invalid!
发表于2023-04-20 10:34:35
0 查看

1 系统环境

硬件环境(Ascend/GPU/CPU): CPU

MindSpore版本: 1.9.0

执行模式(PyNative/ Graph): PyNative

Python版本: 3.7.5

操作系统平台:不限

2 报错信息

2.1  问题描述

针对shape为(1,)的输入,SoftMax的输出结果应该为1,但是MindSpore直接抛出异常

2.2  报错信息

Run op inputs type is invalid!

2.3  脚本代码

import mindspore as ms

print(ms.ops.Softmax(-1)(ms.Tensor(np.ones(1)), ms.float32))

3 根因分析

******此处由用户补充详细的定位过程******

参考API的使用说明

4 解决方案

******此处由用户填写******

包含文字方案和最终脚本代码

支持上传附件

我要发帖子