MindSpore中的ops.cumsum 函数首次计算结果错误
收藏回复举报
MindSpore中的ops.cumsum 函数首次计算结果错误
发表于2023-05-22 19:05:13
0 查看

1 系统环境

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

MindSpore版本: 2.0.0-rc1

执行模式(PyNative/ Graph): 不限

Python版本: 3.9

操作系统平台: ubuntu

2 报错信息

2.1 问题描述

ops.cumsum(ops.arange(1,10).reshape(3,3),axis=1)

-------------------------------------------------------

以下是第一次输出

Tensor(shape=[3, 3], dtype=Int64, value=
[[ 1,  2,  3],
 [ 5,  7,  9],
 [12, 15, 18]])

再次执行上面的代码

ops.cumsum(ops.arange(1,10).reshape(3,3),axis=1)

-------------------------------------------------------

以下是第二次输出

Tensor(shape=[3, 3], dtype=Int64, value=
[[ 1,  3,  6],
 [ 4,  9, 15],
 [ 7, 15, 24]])

实际上第二次输出的结果才是正确的,第一次输出的结果是axis=0的情况。

2.2 报错信息

首次计算结果错误

2.3 脚本代码(代码格式,可上传附件)

见问题描述

3 根因分析

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

4 解决方案

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

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

支持上传附件

本帖最后由 匿名用户2023/05/22 19:41:29 编辑

我要发帖子