!4352 support bool tensor and bool do equal in pynative mode

Merge pull request !4352 from zhangbuxue/support_bool_tensor_and_bool_do_equal
pull/4352/MERGE
mindspore-ci-bot 5 years ago committed by Gitee
commit b337b15c9b

@ -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):

Loading…
Cancel
Save