Browse Source
fix(mge/functional): fix return dtype of comparison function
GitOrigin-RevId: 810e32a829
release-1.10
Megvii Engine Team
3 years ago
1 changed files with 5 additions and 5 deletions
Unified View
Diff Options
+ 5
- 5
imperative/python/megengine/functional/elemwise.py
View File
| @@ -626,7 +626,7 @@ def logaddexp(x: Tensor, y: Tensor) -> Tensor: | |||||
| def equal(x, y): | def equal(x, y): | ||||
| r"""Element-wise `(x == y)`.""" | r"""Element-wise `(x == y)`.""" | ||||
| return _elwise(x, y, mode=Elemwise.Mode.EQ) | |||||
| return x == y | |||||
| def not_equal(x, y): | def not_equal(x, y): | ||||
| @@ -636,22 +636,22 @@ def not_equal(x, y): | |||||
| def less(x, y): | def less(x, y): | ||||
| r"""Element-wise `(x < y)`.""" | r"""Element-wise `(x < y)`.""" | ||||
| return _elwise(x, y, mode=Elemwise.Mode.LT) | |||||
| return x < y | |||||
| def less_equal(x, y): | def less_equal(x, y): | ||||
| r"""Element-wise `(x <= y)`.""" | r"""Element-wise `(x <= y)`.""" | ||||
| return _elwise(x, y, mode=Elemwise.Mode.LEQ) | |||||
| return x <= y | |||||
| def greater(x, y): | def greater(x, y): | ||||
| r"""Element-wise `(x > y)`.""" | r"""Element-wise `(x > y)`.""" | ||||
| return _elwise(y, x, mode=Elemwise.Mode.LT) | |||||
| return x > y | |||||
| def greater_equal(x, y): | def greater_equal(x, y): | ||||
| r"""Element-wise `(x >= y)`.""" | r"""Element-wise `(x >= y)`.""" | ||||
| return _elwise(y, x, mode=Elemwise.Mode.LEQ) | |||||
| return x >= y | |||||
| # other functions | # other functions | ||||