torch.mtia.stream

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

使用上下文管理器 StreamContext 来选择指定的流。

参数

stream (Stream) – 选定的流。如果为None，此管理器将不执行任何操作。

返回类型

StreamContext

..注意:: 在急切模式下，stream 是 Stream 类型的对象，而在 JIT 模式下不支持 torch.mtia.stream。