StreamContext¶ class torch.xpu.StreamContext(stream)[source]¶ 选择给定流的上下文管理器。 在其上下文中排队的全部 XPU 内核都将排队到选定的流上。 参数 Stream (Stream) – 选定的流。如果为 None,则此管理器为无操作。 注意 流是每个设备的。