diff --git a/mindspore/common/tensor.py b/mindspore/common/tensor.py index 733eae4a2d..155f62e5b5 100644 --- a/mindspore/common/tensor.py +++ b/mindspore/common/tensor.py @@ -85,7 +85,9 @@ class Tensor(Tensor_): return False # bool type is not supported for `Equal` operator in backend. if self.dtype == mstype.bool_ or (isinstance(other, Tensor) and other.dtype == mstype.bool_): - return Tensor(np.array(self.asnumpy() == other.asnumpy())) + if isinstance(other, Tensor): + return Tensor(np.array(self.asnumpy() == other.asnumpy())) + return Tensor(np.array(self.asnumpy() == other)) return tensor_operator_registry.get('__eq__')(self, other) def __ne__(self, other):