StreamContext

类torch.cuda.StreamContext(stream)[源代码]

用于选择指定流的上下文管理器。

其上下文中所有排队的 CUDA 内核将被放入选定的流中。

参数

Stream (Stream) – 选择的流。如果该管理器为 None，则它不会执行任何操作。

注意

流是针对每个设备的。