p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!22253 Fix numpy.rollaxis in graph mode 

Merge pull request !22253 from wangrao124/fix_rollaxis
This commit is contained in:
i-robot 2021-08-24 08:36:12 +00:00 committed by Gitee
parent 5c938794ca 7331a5712d
commit 9394686885
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/numpy/array_ops.py
View File

@ -200,7 +200,7 @@ def rollaxis(x, axis, start=0):
axis = _check_axes_range(axis, ndim)
start = _check_start_normalize(start, ndim)
if 0 <= start - axis <= 1:
if start - axis >= 0 and start - axis <= 1:
return x
perm = F.make_range(0, ndim)
new_perm = None