From 2c70842ed9e7cd44b060f4b91f678f62742848df Mon Sep 17 00:00:00 2001 From: jiangjinsheng Date: Wed, 15 Jul 2020 10:44:32 +0800 Subject: [PATCH] fix InvertPermutation error msg --- mindspore/ops/operations/array_ops.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mindspore/ops/operations/array_ops.py b/mindspore/ops/operations/array_ops.py index 5ea52785f6..47df5a361c 100644 --- a/mindspore/ops/operations/array_ops.py +++ b/mindspore/ops/operations/array_ops.py @@ -1018,7 +1018,8 @@ class InvertPermutation(PrimitiveWithInfer): raise ValueError(f'For \'{self.name}\' the input value must be non-Tensor.') for shp in x_shp: if shp != []: - raise ValueError(f'For \'{self.name}\' the rank of input must be 1.') + x_rank = len(np.array(x_value, np.int64).shape) + raise ValueError(f'For \'{self.name}\' the rank of input must be 1, but got {x_rank}.') for i, value in enumerate(x_value): validator.check_value_type("input[%d]" % i, value, [int], self.name) z = [x_value[i] for i in range(len(x_value))]