UniqueInverse¶

Return unique values from x, along with indices, inverse indices, and counts.

Abstract Signature:

UniqueInverse(x: Tensor, size: int, fill_value: Tensor)

API: jax.numpy.unique_inverse

Strategy: Direct Mapping

API: numpy.unique_inverse

Strategy: Direct Mapping