torch.cpu.synchronize

torch.cpu.synchronize(device=None)

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

參數

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

注意：此函數僅為方便編寫與設備無關的程式碼而存在。