torch.cuda.manual_seed_all

torch.cuda.manual_seed_all(seed)

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

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

引數

seed (int) – 期望的種子。