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)