torch.cuda.set_rng_state_all
-
torch.cuda.set_rng_state_all(new_states)[source] -
Устанавливает состояние генератора случайных чисел для всех устройств.
- Параметры:
-
new_states (Итерируемый список torch.ByteTensor) – Желаемое состояние для каждого устройства
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/1.13/generated/torch.cuda.set_rng_state_all.html
