RandIntΒΆ

Returns a tensor filled with random integers generated uniformly between low (inclusive) and high (exclusive).

Abstract Signature:

RandInt(low: int = 0, high: int, size: int)

PyTorch

API: torch.randint
Strategy: Direct Mapping

JAX (Core)

API: jax.random.randint
Strategy: Plugin (inject_prng)

NumPy

API: np.random.randint
Strategy: Direct Mapping

Keras

API: keras.random.randint
Strategy: Direct Mapping

TensorFlow

API: tf.random.uniform
Strategy: Macro 'tf.random.uniform(shape={size}, minval={low}, maxval={high}, dtype=tf.int64)'

Apple MLX

API: mlx.core.random.randint
Strategy: Direct Mapping

Flax NNX

API: jax.random.randint
Strategy: Plugin (inject_prng)