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: