StreamContext

class torch.cuda.StreamContext(stream)

選取給定串流的上下文管理器。

在其上下文中排隊的所有 CUDA 核心都將在選定的串流上排隊。

參數

Stream (Stream) – 選定的串流。如果它是 None，則此管理器是 no-op。

注意

串流是每個裝置的。