tf.random.experimental.create_rng_state

View source on GitHub

Creates a RNG state.

tf.random.experimental.create_rng_state(
    seed, algorithm
)

Args:

• seed: an integer or 1-D tensor.
• algorithm: an integer representing the RNG algorithm.

Returns:

a 1-D tensor whose size depends on the algorithm.