jax.numpy.array_equiv#
- jax.numpy.array_equiv(a1, a2)[source]#
Returns True if input arrays are shape consistent and all elements equal.
LAX-backend implementation of
numpy.array_equiv()
.Original docstring below.
Shape consistent means they are either the same shape, or one input array can be broadcasted to create the same shape as the other one.
- Parameters:
a1 (array_like) – Input arrays.
a2 (array_like) – Input arrays.
- Returns:
out – True if equivalent, False otherwise.
- Return type: