From 95a6ee754d0b9d409b44402bda68d61a7ee350a2 Mon Sep 17 00:00:00 2001 From: Peilin Wang Date: Thu, 3 Dec 2020 16:40:06 -0500 Subject: [PATCH] initial commit --- 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 eecdcd813c..e5e1dcf0f6 100644 --- a/mindspore/ops/operations/array_ops.py +++ b/mindspore/ops/operations/array_ops.py @@ -797,7 +797,7 @@ class GatherV2(PrimitiveWithCheck): def __check__(self, params, indices, axis): validator.check_subclass("params", params['dtype'], mstype.tensor, self.name) validator.check_tensor_dtype_valid("indices", indices['dtype'], mstype.int_type, self.name) - validator.check_subclass("axis", axis['dtype'], [mstype.tensor, mstype.int_], self.name) + validator.check_subclass("axis", axis['dtype'], [mstype.int_], self.name) class SparseGatherV2(GatherV2):