torch.cuda.set_rng_state_all

torch.cuda.set_rng_state_all(new_states)

Set the random number generator state of all devices.

Parameters

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