RandIntLike

Returns a tensor with the same shape as Tensor input filled with random integers.

Abstract Signature:

RandIntLike(input: Tensor, low: int = 0, high: int)

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLX

PyTorch

API: torch.randint_like

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: jax.random.randint

Strategy: Plugin (inject_prng)

NumPy

API: np.random.randint

Strategy: Macro 'np.random.randint({low}, {high}, size={input}.shape)'

Keras

API: keras.random.randint

Strategy: Macro 'keras.random.randint({input}.shape, minval={low}, maxval={high})'

TensorFlow

API: tf.random.uniform

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

Apple MLX

API: mlx.core.random.randint

Strategy: Macro 'mlx.core.random.randint({low}, {high}, {input}.shape)'