Bucketize

Returns the indices of the buckets to which each value in the input belongs.

Abstract Signature:

Bucketize(input: Tensor, boundaries: Tensor, out_int32: bool = False, right: bool = False)

PyTorch

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

JAX (Core)

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

NumPy

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

TensorFlow

API: tf.searchsorted
Strategy: Direct Mapping
Official Docs ↗