StreamContext
- class torch.cpu.StreamContext(stream)[source]
Context-manager that selects a given stream.
N.B. This class only exists to facilitate device-agnostic code
Context-manager that selects a given stream.
N.B. This class only exists to facilitate device-agnostic code