torch.manual_seed() 介绍
-
概述
用于设置CPU生成随机数的种子。返回一个torch.Generator对象。此时接下来运行随机函数生成的随机数都不会发生变化,方便论文复现结果。
-
语法
torch.manual_seed(seed) -
参数
seed (int) – The desired seed. Value must be within the inclusive range [-0x8000_0000_0000_0000, 0xffff_ffff_ffff_ffff]. Otherwise, a RuntimeError is raised. Negative inputs are remapped to positive values with the formula 0xffff_ffff_ffff_ffff + seed.
测试一:不设置seed,生成随机数
代码如下:
import torch
print(torch.rand(1)) # 随机生成[0, 1)的数
每次运行结果都不同:
tensor([0.1580])
tensor([0.9103])
本文详细介绍了torch.manual_seed()的作用,通过多个测试案例展示了设置和不设置seed时,生成随机数的不同行为。设置seed可以确保在复现实验结果时,每次运行的随机数序列一致。不设置seed时,每次运行会得到不同的随机数。理解这一概念对于深度学习模型的可重复性至关重要。
最低0.47元/天 解锁文章
442

被折叠的 条评论
为什么被折叠?



