torch.cpu.stream

torch.cpu.stream(stream)

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

N.B. This function only exists to facilitate device-agnostic code

Return type

AbstractContextManager