torch.accelerator.set_stream

torch.accelerator.set_stream(stream)

將當前流設定為給定的流。

引數

stream (torch.Stream) – 給定的流，必須與當前 accelerator 裝置型別匹配。

注意

此函式會將當前裝置索引設定為給定流的裝置索引。