span> <span class="mi">1</span><span class="p">],</span> <span class="n">edge_input_dim</span><span class="p">,</span> <span class="n">num_head</span><span class="p">,</span>
                                                         <span class="n">negative_slope</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="k">if</span> <span class="n">readout</span> <span class="o">==</span> <span cla