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

fix InvertPermutation error msg

This commit is contained in:
jiangjinsheng 2020-07-15 10:44:32 +08:00
parent bdcc607b1a
commit 2c70842ed9
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ops/operations/array_ops.py
View File

@ -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))]