PyTorch torch.set_rng_state 関数
Pytorch torch リファレンスマニュアルtorch.set_rng_statePyTorch で乱数生成器の状態を設定するための関数です。
関数定義
torch.set_rng_state(new_state)
パラメータ説明
new_state- 新しい乱数生成器の状態（Tensor型）
使用例
実例
import torch
# 現在の乱数生成器の状態を保存
saved_state = torch.get_rng_state()
# 乱数シードを設定
torch.manual_seed(123)
# いくつかの乱数を生成
print("1回目の乱数:", torch.rand(3))
# 以前に保存した状態を復元
torch.set_rng_state(saved_state)
# 乱数を再生成。以前と同じになるはず
print("復元後の乱数:", torch.rand(3))
# 現在の乱数生成器の状態を保存
saved_state = torch.get_rng_state()
# 乱数シードを設定
torch.manual_seed(123)
# いくつかの乱数を生成
print("1回目の乱数:", torch.rand(3))
# 以前に保存した状態を復元
torch.set_rng_state(saved_state)
# 乱数を再生成。以前と同じになるはず
print("復元後の乱数:", torch.rand(3))
その他の拡張