torch.cpu.set_device

torch.cpu.set_device(device)

设置当前设备，在 CPU 中我们不做任何事情。

注意。此函数仅存在以方便设备无关的代码。