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