流式上下文 ¶

class torch.cuda.StreamContext(stream)

选择给定流的上下文管理器。

所有在该上下文中排队的 CUDA 内核都将被添加到选定的流中。

参数:

流（Stream）- 选定的流。如果为 None ，则此管理器为空操作。

注意

流是按设备划分的。