jax.numpy.unique_inverse# jax.numpy.unique_inverse(x, /)[source]# Parameters: x (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – Return type: _UniqueInverseResult