get_device
==========

.. currentmodule:: lib.torch_utils

.. autofunction:: get_device