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

fixed InvertPermutation

This commit is contained in:
jiangjinsheng 2020-05-27 09:46:38 +08:00
parent a45dfca142
commit 4ff95eab69
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/ops/operations/array_ops.py
View File

@ -980,7 +980,7 @@ class InvertPermutation(PrimitiveWithInfer):
validator.check_value_type("shape", x_shp, [tuple, list], self.name) validator.check_value_type("shape", x_shp, [tuple, list], self.name)
if mstype.issubclass_(x['dtype'], mstype.tensor): if mstype.issubclass_(x['dtype'], mstype.tensor):
validator.check('x dimension', len(x_shp), '', 1, Rel.EQ, self.name) validator.check('x dimension', len(x_shp), '', 1, Rel.EQ, self.name)
validator.check_type_same({'x dtype': x['dtype']}, mstype.int_type, self.name) validator.check_tensor_type_same({'x dtype': x['dtype']}, mstype.int_type, self.name)
x_value = [int(i) for i in x_value.asnumpy()] x_value = [int(i) for i in x_value.asnumpy()]
z = [x_value[i] for i in range(len(x_value))] z = [x_value[i] for i in range(len(x_value))]
z.sort() z.sort()