torch.cuda.set_rng_state

torch.cuda.set_rng_state(new_state, device='cuda') [source]

Устанавливает состояние генератора случайных чисел для указанной видеокарты.

Параметры:

• new_state (torch.ByteTensor) – Желаемое состояние
• device (torch.device или int, необязательно) – Устройство для установки состояния генератора случайных чисел. По умолчанию: 'cuda' (т.е. torch.device('cuda'), текущее устройство CUDA).