torch.cpu.current_stream¶ torch.cpu.current_stream(device=None)[源代码]¶ 返回给定设备的当前选定的Stream。 Parameters 设备 (torch.device 或 int, 可选) – 忽略。 Return type 流 注意：此函数仅用于实现设备无关的代码
