torch.cpu.current_stream

torch.cpu.current_stream(device=None)

返回给定设备当前选定的 Stream 。

参数:

device (torch.device 或 int，可选) – 被忽略。

返回类型:

流

注意。此函数仅存在以方便设备无关的代码