StreamContext

class torch.xpu.StreamContext(stream)

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

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

引數

Stream (Stream) – 選定的流。如果為 None，則此管理器為空操作。

注意

流是按裝置劃分的。