torch.cpu.set_device

torch.cpu.set_device(device) [исходный код]

Устанавливает текущее устройство; для CPU ничего не делает.

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