Browse Source

fix(mge/functional): fix return dtype of comparison function

GitOrigin-RevId: 810e32a829
release-1.10
Megvii Engine Team 3 years ago
parent
commit
dbce6526d6
1 changed files with 5 additions and 5 deletions
  1. +5
    -5
      imperative/python/megengine/functional/elemwise.py

+ 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):
r"""Element-wise `(x == y)`."""
return _elwise(x, y, mode=Elemwise.Mode.EQ)
return x == y


def not_equal(x, y):
@@ -636,22 +636,22 @@ def not_equal(x, y):

def less(x, y):
r"""Element-wise `(x < y)`."""
return _elwise(x, y, mode=Elemwise.Mode.LT)
return x < y


def less_equal(x, y):
r"""Element-wise `(x <= y)`."""
return _elwise(x, y, mode=Elemwise.Mode.LEQ)
return x <= y


def greater(x, y):
r"""Element-wise `(x > y)`."""
return _elwise(y, x, mode=Elemwise.Mode.LT)
return x > y


def greater_equal(x, y):
r"""Element-wise `(x >= y)`."""
return _elwise(y, x, mode=Elemwise.Mode.LEQ)
return x >= y


# other functions


Loading…
Cancel
Save