torch.cuda.manual_seed_all

torch.cuda.manual_seed_all(seed)

設定所有 GPU 上生成隨機數的種子。

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

引數

seed (int) – 期望的種子。