torch.set_rng_state#
- torch.set_rng_state(new_state)[source]#
Sets the random number generator state.
Note
This function only works for CPU. For CUDA, please use
torch.manual_seed(), which works for both CPU and CUDA.- Parameters
new_state (torch.ByteTensor) – The desired state