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