torch.xpu.seed¶
- torch.xpu.seed()[源][源]¶
為當前 GPU 設定用於生成隨機數的種子為一個隨機數。
如果 XPU 不可用,呼叫此函式是安全的;在這種情況下,它會被靜默忽略。
警告
如果您正在處理多 GPU 模型,此函式將只在一塊 GPU 上初始化種子。要初始化所有 GPU,請使用
seed_all()。
為當前 GPU 設定用於生成隨機數的種子為一個隨機數。
如果 XPU 不可用,呼叫此函式是安全的;在這種情況下,它會被靜默忽略。
警告
如果您正在處理多 GPU 模型,此函式將只在一塊 GPU 上初始化種子。要初始化所有 GPU,請使用 seed_all()。