torch.set_rng_state
-
torch.set_rng_state(new_state)
[source] -
Sets the random number generator state.
- Parameters
-
new_state (torch.ByteTensor) – The desired state
© 2019 Torch Contributors
Licensed under the 3-clause BSD License.
https://pytorch.org/docs/1.8.0/generated/torch.set_rng_state.html