快速鍵

torch.cuda.set_rng_state_all

torch.cuda.set_rng_state_all(new_states)[原始碼][原始碼]

設定所有裝置的亂數產生器狀態。

參數

new_states (torch.ByteTensorIterable) – 每個裝置的所需狀態。

文件

取得 PyTorch 的完整開發者文件

檢視文件

教學

取得初學者和進階開發者的深入教學

檢視教學

資源

尋找開發資源並獲得問題解答

檢視資源