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
Sets the random number generator state.
new_state (torch.ByteTensor) – The desired state