快捷方式

torch.xpu.stream

torch.xpu.stream(stream)[原始碼][原始碼]

封裝 StreamContext 上下文管理器,用於選擇指定的 stream。

參數

stream (Stream) – 選取的 stream。如果為 None,則此管理器不執行任何操作。

回傳類型

StreamContext

文件

存取 PyTorch 的完整開發者文件

檢視文件

教學

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

檢視教學

資源

尋找開發資源並取得您的問題解答

檢視資源