Digitize

Return the indices of the bins to which each value in input array belongs.

Abstract Signature:

Digitize(x: Tensor, bins: Tensor, right: bool = False)

PyTorch

API: torch.bucketize
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

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

NumPy

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

Keras

API: keras.ops.digitize
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.experimental.numpy.digitize
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.digitize
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

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

PaxML / Praxis

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