unique_inverse ============== Returns the unique elements of an input array ``x`` and the indices from the set of unique elements that reconstruct ``x``. **Abstract Signature:** ``unique_inverse(x: array)`` .. raw:: html

JAX (Core)

API: jax.numpy.unique_inverse
Strategy: Direct Mapping

NumPy

API: numpy.unique_inverse
Strategy: Direct Mapping

Flax NNX

API: jax.numpy.unique_inverse
Strategy: Direct Mapping