View source on GitHub |
Splits a seed into n
derived seeds.
tfp.substrates.numpy.random.split_seed(
seed, n=2, salt=None, name=None
)
See https://github.com/tensorflow/probability/blob/main/PRNGS.md
for details.
Args:
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. In TensorFlow, if n
is an integer, this
function returns a list of seeds and otherwise returns a Tensor
of
seeds. In JAX, this function always returns an array of seeds.
salt: Optional str
salt to mix with the seed.
name: Optional name to scope related ops.