torch.xpu.manual_seed

torch.xpu.manual_seed(seed)

為當前 GPU 設定用於生成隨機數的種子。

如果 XPU 不可用，呼叫此函式是安全的；在這種情況下，它會被靜默忽略。

引數

seed (int) – 所需的種子。

警告

如果您使用的是多 GPU 模型，此函式不足以確保確定性。要為所有 GPU 設定種子，請使用 manual_seed_all()。