torch.xpu.set\_device ===================== .. currentmodule:: torch.xpu .. autofunction:: set_device