">1</span>
        <span class="k">return</span> <span class="n">torch</span><span class="o">.</span><span class="n">cat</span><span class="p">(</span><span class="n">result</span><span class="p">,</span> <span class="n">last_dim</span><span class="p">),</span> <span class="n">args</span><span class="p">[</span><span class="mi">1</span><span class="p">]</span>  <span class="c1">#concat on the last dimension</span></div></div>

</pre></div>

           </div>
           
          </div>
          <footer>
 