jax.numpy.array_equiv

Contents

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:

bool