torch.cpu.current_stream
- torch.cpu.current_stream(device=None)[source]
Returns the currently selected
Stream
for a given device.- Parameters
device (torch.device or int, optional) – Ignored.
- Return type
N.B. This function only exists to facilitate device-agnostic code