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)

JAX (Core)

API: jax.numpy.unique_inverse
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.unique_inverse
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: jax.numpy.unique_inverse
Strategy: Direct Mapping
Official Docs ↗