torch.cpu.stream

torch.cpu.stream(stream)

围绕 StreamContext 的包装器，用于选择指定的流。

注意：此函数仅用于方便设备无关的代码

返回类型

AbstractContextManager