tfp.experimental.substrates.jax.random.split_seed

Splits a seed into n derived seeds.

seed The seed to split; may be an int, an (int, int) tuple, or a Tensor. int seeds are converted to Tensor seeds using tf.random.stateless_uniform stateful sampling. Tuples are converted to Tensor.
n The number of splits to return.
salt Optional str salt to mix with the seed.
name Optional name to scope related ops.

seeds If n is a Python int, a tuple of seed values is returned. If n is an int Tensor, a single Tensor of shape [n, 2] is returned. A single such seed is suitable to pass as the seed argument of the tf.random.stateless_* ops.