torch.cpu.synchronize

torch.cpu.synchronize(device=None) [source]

Ожидает завершения всех ядер во всех потоках на устройстве CPU.

Параметры

device (torch.device или int, необязательно) – игнорируется, есть только одно устройство CPU.

Примечание. Эта функция существует только для обеспечения кода, не зависящего от устройства.