jax.numpy.equal¶
-
jax.numpy.
equal
(x1, x2)¶ Return (x1 == x2) element-wise.
LAX-backend implementation of
equal()
. Original docstring below.equal(x1, x2, /, out=None, *, where=True, casting=’same_kind’, order=’K’, dtype=None, subok=True[, signature, extobj])
- Parameters
x2 (x1,) – Input arrays. If
x1.shape != x2.shape
, they must be broadcastable to a common shape (which becomes the shape of the output).- Returns
out – Output array, element-wise comparison of x1 and x2. Typically of type bool, unless
dtype=object
is passed. This is a scalar if both x1 and x2 are scalars.- Return type
ndarray or scalar
See also
not_equal()
,greater_equal()
,less_equal()
,greater()
,less()
Examples
>>> np.equal([0, 1, 3], np.arange(3)) array([ True, True, False])
What is compared are values, not types. So an int (1) and an array of length one can evaluate as True:
>>> np.equal(1, np.ones(1)) array([ True])