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))