torch.cpu.set_device
- torch.cpu.set_device(device)[source]
Sets the current device, in CPU we do nothing.
N.B. This function only exists to facilitate device-agnostic code
Sets the current device, in CPU we do nothing.
N.B. This function only exists to facilitate device-agnostic code