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