捷徑

torch.xpu.set_stream

torch.xpu.set_stream(stream)[來源][來源]
設定目前 stream。這是設定 stream 的包裝函式 API。

不鼓勵使用此函數,而建議使用 stream 環境管理器。

參數

stream (Stream) – 選取的 stream。如果此引數為 None,則此函數為空操作。

文件

存取 PyTorch 的完整開發者文件

檢視文件

教學

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

檢視教學

資源

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

檢視資源