torch.cpu.synchronize

torch.cpu.synchronize(device=None)

等待 CPU 裝置上所有流中的所有核心完成。

引數

device (torch.device 或 int, 可選) ：忽略，因為只有一個 CPU 裝置。

注意：此函式僅用於方便編寫與裝置無關的程式碼。