torch.cpu.stream

torch.cpu.stream(stream)

Wrapper around the Context-manager StreamContext that selects a given stream.

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

返回类型

AbstractContextManager