n">batch_max_points</span> <span class="o">==</span> <span class="mi">0</span><span class="p">:</span>
        <span class="c1"># if we are here, batch consists only of empty diagrams.</span>
        <span class="n">batch_max_points</span> <span class="o">=</span> <span class="mi">1</span>

    <span class="c1"># This will later be used to set the dummy points to zero in the output.</span>
    <span class="n">not_dummy_points</span> <span class="o">=</span> <span class="n">torch</span><span class="o">.</spa