torch.set_rng_state¶
- torch.set_rng_state(new_state)[來源][來源]¶
設定亂數產生器狀態。
注意
此函數僅適用於 CPU。對於 CUDA,請使用
torch.manual_seed()
,它同時適用於 CPU 和 CUDA。- 參數
new_state (torch.ByteTensor) – 期望的狀態
設定亂數產生器狀態。
注意
此函數僅適用於 CPU。對於 CUDA,請使用 torch.manual_seed()
,它同時適用於 CPU 和 CUDA。
new_state (torch.ByteTensor) – 期望的狀態