From 5ab32c33e45d0e8405b0c183d775a1ed8abb7aaa Mon Sep 17 00:00:00 2001 From: buxue Date: Wed, 27 May 2020 17:21:26 +0800 Subject: [PATCH] support tensor set item the number value type is similar as tensor dtype not same --- mindspore/ops/composite/multitype_ops/_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/ops/composite/multitype_ops/_utils.py b/mindspore/ops/composite/multitype_ops/_utils.py index 9717ddc15e6..ee914aeeb2e 100644 --- a/mindspore/ops/composite/multitype_ops/_utils.py +++ b/mindspore/ops/composite/multitype_ops/_utils.py @@ -280,7 +280,7 @@ def check_value_elements(data_dtype, types): else: raise TypeError(f"For '{TENSOR_SETITEM}', the data type of {i}th tensor '{ele_dtype}' " f"in value tuple is not consistent with origin tensor data type '{data_dtype}'.") - elif mstype.issubclass_(ele, data_dtype): + elif mstype.dtype_to_pytype(ele) == mstype.dtype_to_pytype(data_dtype): scalars_number += 1 else: raise TypeError(f"For '{TENSOR_SETITEM}', the {i}th element type '{ele}' in "