捷徑

torch.mps.get_rng_state

torch.mps.get_rng_state(device='mps')[source][source]

以 ByteTensor 形式傳回隨機數字產生器狀態。

參數

device (torch.deviceint, optional) – 要傳回 RNG 狀態的裝置。預設值:'mps'(即 torch.device('mps'),目前的 MPS 裝置)。

傳回類型

Tensor

文件

取得 PyTorch 的完整開發者文件

檢視文件

教學

取得為初學者和進階開發人員提供的深入教學

檢視教學

資源

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

檢視資源