Python PyTorch Generator.manual_seed用法及代码示例

本文简要介绍python语言中 torch.Generator.manual_seed 的用法。

用法:

manual_seed(seed) → Generator

参数：

seed(int) -所需的种子。值必须在 [-0x8000_0000_0000_0000, 0xffff_ffff_ffff_ffff] 范围内。否则，会引发 RuntimeError。使用公式 0xffff_ffff_ffff_ffff + seed 将负输入重新映射为正值。

返回：

一个 torch.Generator 对象。

返回类型：

生成器

设置生成随机数的种子。返回一个 torch.Generator 对象。建议设置大种子，即 0 位和 1 位平衡良好的数字。避免种子中有许多 0 位。

例子：

>>> g_cpu = torch.Generator()
>>> g_cpu.manual_seed(2147483647)