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 "