torch.cpu.synchronize

torch.cpu.synchronize(device=None)

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

引數

device (torch.device 或 int, optional) – 忽略，只有一個 CPU 裝置。

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