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
torch.cuda.
set_rng_state_all
(new_states)[source]Sets the random number generator state of all devices.
new_states (Iterable of torch.ByteTensor) – The desired state for each device