torch.mtia.stream ¬

torch.mtia.stream(stream)[来源][来源] ¬

包装上下文管理器 StreamContext，该上下文管理器选择指定的流。

参数:

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

返回类型:

Stream 上下文

注意

在急切模式下，流是 Stream 类的类型，而在即时编译（JIT）模式下不支持 torch.mtia.stream