torch.set_rng_state
-
torch.set_rng_state(new_state)[исходный код] -
Задаёт состояние генератора случайных чисел.
Примечание
Эта функция работает только для CPU. Для CUDA используйте
torch.manual_seed(), которая работает и для CPU, и для CUDA.- Параметры:
-
new_state (torch.ByteTensor) – Требуемое состояние
© 2026, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://docs.pytorch.org/docs/2.14/generated/torch.set_rng_state.html
