torch.cuda.set_rng_state_all
- torch.cuda.set_rng_state_all(new_states)[source]
Sets the random number generator state of all devices.
- Parameters
new_states (Iterable of torch.ByteTensor) – The desired state for each device
Sets the random number generator state of all devices.
new_states (Iterable of torch.ByteTensor) – The desired state for each device