jax.numpy.array_equal#

jax.numpy.array_equal(a1, a2, equal_nan=False)[source]#

True if two arrays have the same shape and elements, False otherwise.

LAX-backend implementation of numpy.array_equal().

Original docstring below.

Parameters

  • a1 (array_like) – Input arrays.

  • a2 (array_like) – Input arrays.

  • equal_nan (bool) – Whether to compare NaN’s as equal. If the dtype of a1 and a2 is complex, values will be considered equal if either the real or the imaginary component of a given value is nan.

Returns

b – Returns True if the arrays are equal.

Return type

bool