PyTorch torch.get_rng_state 函数
Pytorch torch 参考手册torch.get_rng_state 是 PyTorch 中用于返回随机数生成器状态的函数。
函数定义
torch.get_rng_state()
使用示例
实例
import torch
# 获取当前随机数生成器状态
rng_state = torch.get_rng_state()
print("随机数生成器状态:", rng_state)
print("状态类型:", type(rng_state))
print("状态大小:", rng_state.numel())
# 获取当前随机数生成器状态
rng_state = torch.get_rng_state()
print("随机数生成器状态:", rng_state)
print("状态类型:", type(rng_state))
print("状态大小:", rng_state.numel())