p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

support bool tensor and bool do equal

This commit is contained in:
buxue 2020-08-12 18:12:33 +08:00
parent 363fbb7a79
commit 9c7cdcf1e9
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/common/tensor.py
View File

@ -85,7 +85,9 @@ class Tensor(Tensor_):
return False return False
# bool type is not supported for `Equal` operator in backend. # bool type is not supported for `Equal` operator in backend.
if self.dtype == mstype.bool_ or (isinstance(other, Tensor) and other.dtype == mstype.bool_): 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) return tensor_operator_registry.get('__eq__')(self, other)
def __ne__(self, other): def __ne__(self, other):