tensorplay.cuda.random.set_rng_state_all

tensorplay.cuda.random.set_rng_state_all(new_states: Iterable[Tensor]) → None[source]

Set the random number generator state of all devices.

Parameters:

new_states (Iterable of ByteTensor) – The desired state for each device.