torch.accelerator.set\_device\_idx ================================== .. currentmodule:: torch.accelerator .. autofunction:: set_device_idx