forked from mindspore-Ecosystem/mindspore
fixed InvertPermutation
This commit is contained in:
parent
a45dfca142
commit
4ff95eab69
mindspore/ops/operations
|
@ -980,7 +980,7 @@ class InvertPermutation(PrimitiveWithInfer):
|
|||
validator.check_value_type("shape", x_shp, [tuple, list], self.name)
|
||||
if mstype.issubclass_(x['dtype'], mstype.tensor):
|
||||
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()]
|
||||
z = [x_value[i] for i in range(len(x_value))]
|
||||
z.sort()
|
||||
|
|
Loading…
Reference in New Issue