Rate this Page
★ ★ ★ ★ ★

torch.cuda.get_rng_state_all#

torch.cuda.get_rng_state_all()[source]#

Return a list of ByteTensor representing the random number states of all devices.

Return type:

list[Tensor]

On this page
Show Source
PyTorch Libraries