jax.numpy.array_equal
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