torch.xpu.stream¶ torch.xpu.stream(stream)[原始碼][原始碼]¶ 封裝 StreamContext 上下文管理器,用於選擇指定的 stream。 參數 stream (Stream) – 選取的 stream。如果為 None,則此管理器不執行任何操作。 回傳類型 StreamContext