torch.xpu.set_rng_state

torch.xpu.set_rng_state(new_state, device='xpu') [исходный код]

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

Параметры:

• new_state (torch.ByteTensor) – Требуемое состояние
• device (torch.device or int, необязательно) – Устройство, для которого нужно установить состояние RNG. По умолчанию: 'xpu' (т. е. torch.device('xpu'), текущее устройство XPU).