unique_allΒΆ

Returns the unique elements of an input array x, the first occurring indices for each unique element in x, the indices from the set of unique elements that reconstruct x, and the corresponding counts for each unique element in x.

Abstract Signature:

unique_all(x: array)

JAX (Core)

API: jax.numpy.unique_all
Strategy: Direct Mapping

NumPy

API: numpy.unique_all
Strategy: Direct Mapping

Flax NNX

API: jax.numpy.unique_all
Strategy: Direct Mapping