Shortcuts

torch.cpu.stream

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

围绕上下文管理器 StreamContext 的包装器,用于选择给定的流。

注意:此函数仅用于实现设备无关的代码

Return type

抽象上下文管理器