torch.cpu.stream

torch.cpu.stream(stream)

包装 Context-manager StreamContext 的包装器，用于选择指定的流。

注意。此函数仅存在以方便设备无关的代码

返回类型:

抽象上下文管理器