From 7331a5712d08cb68c31bbc6bfbf6e42f2c65204c Mon Sep 17 00:00:00 2001 From: wangrao124 Date: Mon, 23 Aug 2021 17:47:08 +0800 Subject: [PATCH] fix numpy.rollaxis in graph mode --- mindspore/numpy/array_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/numpy/array_ops.py b/mindspore/numpy/array_ops.py index 8fecaa811ee..92189ae52c0 100644 --- 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