Unique

Return the unique values from an array.

Abstract Signature:

Unique(ar: Tensor, return_index: bool = False, return_inverse: bool = False, return_counts: bool = False, axis: int, equal_nan: bool = True, size: int, fill_value: Tensor, sorted: bool = True)

PyTorchJAX (Core)NumPyTensorFlowApple MLXFlax NNX

PyTorch

API: torch.unique

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.numpy.unique

Strategy: Direct Mapping

Official Docs ↗

NumPy

API: numpy.unique

Strategy: Direct Mapping

Official Docs ↗

TensorFlow

API: tf.unique

Strategy: Direct Mapping

Official Docs ↗

Apple MLX

API: mlx.core.unique

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: jax.numpy.unique

Strategy: Direct Mapping

Official Docs ↗