StreamContext

class torch.cuda.StreamContext(stream)

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

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

引數

Stream (Stream) – 所選流。如果它是 None，則此管理器是一個空操作。

注意

流是按裝置劃分的。