快捷方式

torch.xpu.set_stream

torch.xpu.set_stream(stream)[source][source]
設定當前流。這是一個用於設定流的包裝 API。

不鼓勵使用此函式,推薦使用 stream 上下文管理器。

引數

stream (Stream) – 選定的流。如果此引數為 None,則此函式不執行任何操作 (no-op)。

文件

查閱 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深度教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源