CodeNodeX AI
Sign upWrite
TUTORIALAI & dataFETCHED

PyTorch torch.set_rng_state 函数

PyTorch torch.set_rng_state 函数 Pytorch torch 参考手册 torch.set_rng_state 是 PyTorch 中用于设置随机数生成器状态的函数。 函数定义 torch.set_rng_state(new_state) 参数说明 new_state - 新的随机数生成器状态（Tensor类型） 使用示例 实例 [mycode4 type='python&..

Updated 7/16/2026
PyTorch

Tutorial detail pages default to noindex.

PyTorch torch.set_rng_state 函数

Pytorch torch 参考手册 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))

Pytorch torch 参考手册 Pytorch torch 参考手册