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