UniqueInverseΒΆ Find the unique elements of x and indices to reconstruct x. Abstract Signature: UniqueInverse(x: Array) JAX (Core)NumPy JAX (Core)API: jnp.unique_inverseStrategy: Direct MappingOfficial Docs βNumPyAPI: numpy.unique_inverseStrategy: Direct MappingOfficial Docs β