tf.random.create_rng_state

View source on GitHub

Creates a RNG state from an integer or a vector.

tf.random.create_rng_state(
    seed, alg
)

Example:

tf.random.create_rng_state(
    1234, "philox")
array([1234,    0,    0])
tf.random.create_rng_state(
    [12, 34], "threefry")
array([12, 34])

seed an integer or 1-D numpy array.
alg the RNG algorithm. Can be a string, an Algorithm or an integer.

a 1-D numpy array whose size depends on the algorithm.