support bool tensor and bool do equal

5 years ago · 9c7cdcf1e9
parent 363fbb7a79
commit 9c7cdcf1e9
1 changed files with 3 additions and 1 deletions
--- 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):