n>

<span class="kn">import</span> <span class="nn">warnings</span>


<span class="c1"># region helper functions</span>


<div class="viewcode-block" id="prepare_batch"><a class="viewcode-back" href="../../../nn.html#torchph.nn.slayer.prepare_batch">[docs]</a><span class="k">def</span> <span class="nf">prepare_batch</span><span class="p">(</span>
        <span class="n">batch</span><span class="p">:</span> <span class="n">List</span><span class="p">[</span><span class="n">Tensor</span><span class="p">],</sp