torch.xpu.manual_seed

torch.xpu.manual_seed(seed)[源代码]

为当前GPU生成随机数设置种子。

如果 XPU 不可用，调用此函数是安全的；此时该函数会被静默忽略。

参数

seed (int) – 指定的随机种子值。

警告

如果你正在处理一个多GPU模型，仅使用这个函数无法获得确定性。为了为所有GPU设置种子，请使用manual_seed_all()。