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

!9474 SparseGatherV2: Raise error if tensor is provided as input for axis 

From: @peilin-wang
Reviewed-by: @tom__chen,@robingrosman
Signed-off-by: @robingrosman
This commit is contained in:
mindspore-ci-bot 2020-12-05 05:28:36 +08:00 committed by Gitee
parent b793c7b291 95a6ee754d
commit a5fd09448f
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

@ -807,7 +807,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):