From 4ff95eab69ba96e89cf8dfb11542a67bcd211956 Mon Sep 17 00:00:00 2001 From: jiangjinsheng Date: Wed, 27 May 2020 09:46:38 +0800 Subject: [PATCH] fixed InvertPermutation --- mindspore/ops/operations/array_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ops/operations/array_ops.py b/mindspore/ops/operations/array_ops.py index 1b33533dc4..757335e293 100644 --- a/mindspore/ops/operations/array_ops.py +++ b/mindspore/ops/operations/array_ops.py @@ -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()