fix numpy.rollaxis in graph mode

2021-08-23 17:47:08 +08:00 · 2021-08-23 17:47:08 +08:00 · 7331a5712d
parent 4661b47b52
commit 7331a5712d
1 changed files with 1 additions and 1 deletions
--- a/mindspore/numpy/array_ops.py
+++ b/mindspore/numpy/array_ops.py
@ -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