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) – 期望的状态