torch.mtia.set\_device
======================

.. currentmodule:: torch.mtia

.. autofunction:: set_device