get_device">
<a class="viewcode-back" href="../../../api/utils.html#labml.utils.pytorch.get_device">[docs]</a>
<span class="k">def</span> <span class="nf">get_device</span><span class="p">(</span><span class="n">module</span><span class="p">:</span> <span class="n">torch</span><span class="o">.</span><span class="n">nn</span><span class="o">.</span><span class="n">Module</span><span class="p">):</span>
<span class="w">    </span><span class="sd">&quot;&quot;&quot;</span>
<span class="sd">    Get the device 