torch.cuda.set_rng_state_all¶ torch.cuda.set_rng_state_all(new_states)[原始碼][原始碼]¶ 設定所有裝置的亂數產生器狀態。 參數 new_states (torch.ByteTensor 的 Iterable) – 每個裝置的所需狀態。