Rate this Page

torch.get_rng_state#

torch.get_rng_state()[source]#

Returns the random number generator state as a torch.ByteTensor.

Note

The returned state is for the default generator on CPU only.

See also: torch.random.fork_rng().

Return type

Tensor