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 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