快捷鍵

torch.xpu.get_rng_state

torch.xpu.get_rng_state(device='xpu')[原始碼][原始碼]

以 ByteTensor 形式傳回指定 GPU 的隨機數產生器狀態。

參數

device (torch.deviceint, 選用) – 要傳回 RNG 狀態的裝置。預設值:'xpu' (即,torch.device('xpu'),目前的 XPU 裝置)。

回傳類型

Tensor

警告

此函式會預先初始化 XPU。

文件

存取 PyTorch 的全面開發者文件

檢視文件

教學

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

檢視教學

資源

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

檢視資源