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)
JAX (Core)
API:
jax.random.randintStrategy: Plugin (inject_prng)
TensorFlow
API:
tf.random.uniformStrategy: Macro 'tf.random.uniform(shape={size}, minval={low}, maxval={high}, dtype=tf.int64)'
Flax NNX
API:
jax.random.randintStrategy: Plugin (inject_prng)