torch.accelerator.set\_device\_index ==================================== .. currentmodule:: torch.accelerator .. autofunction:: set_device_index