><span class="p">,</span> <span class="n">edge_input_dim</span><span class="p">,</span> <span class="p">[</span><span class="n">hidden_dim</span><span class="p">]</span> <span class="o">*</span> <span class="p">(</span><span class="n">num_mlp_layer</span> <span class="o">-</span> <span class="mi">1</span><span class="p">),</span>
                                           <span class="n">batch_norm</span><span class="p">,</span> <span class="n">activation</span><span class="p">)</span>
        <span class="