torch.cuda.set_stream

torch.cuda.set_stream(stream)

設定目前的 stream。這是一個用來設定 stream 的包裝 API。

不建議使用此函式，建議改用 stream 上下文管理器。

參數

stream (Stream) – 選取的串流。如果此引數為 None，則此函式不執行任何動作。