Функция PyTorch torch.set_rng_state
Справочное руководство PyTorch torch
torch.set_rng_stateЭто функция в PyTorch, используемая для установки состояния генератора случайных чисел.
Определение функции
torch.set_rng_state(new_state)
Описание параметра
new_state- Новое состояние генератора случайных чисел (тип Tensor)
Пример использования
Пример
import torch
# Сохранить текущее состояние генератора случайных чисел
saved_state = torch.get_rng_state()
# Установить случайное зерно
torch.manual_seed(123)
# Сгенерировать несколько случайных чисел
print("Первые случайные числа:", torch.rand(3))
# Восстановить ранее сохранённое состояние
torch.set_rng_state(saved_state)
# Заново сгенерировать случайные числа; должны совпасть с предыдущими
print("Случайные числа после восстановления:", torch.rand(3))
# Сохранить текущее состояние генератора случайных чисел
saved_state = torch.get_rng_state()
# Установить случайное зерно
torch.manual_seed(123)
# Сгенерировать несколько случайных чисел
print("Первые случайные числа:", torch.rand(3))
# Восстановить ранее сохранённое состояние
torch.set_rng_state(saved_state)
# Заново сгенерировать случайные числа; должны совпасть с предыдущими
print("Случайные числа после восстановления:", torch.rand(3))
Другие расширения