Shortcuts

torch.xpu.stream

torch.xpu.stream(stream)[源代码]

围绕选择给定流的上下文管理器 StreamContext 进行包装。

Parameters

() – 选择的流。如果它是 None,则此管理器不执行任何操作。

Return type

流上下文

优云智算