unique_counts

Returns the unique elements of an input array x and the corresponding counts for each unique element in x.

Abstract Signature:

unique_counts(x: array)

JAX (Core)

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

NumPy

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

Flax NNX

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