lass="n">input_dim</span><span class="p">,</span> <span class="n">hidden_dims</span><span class="p">,</span> <span class="n">edge_input_dim</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span> <span class="n">num_head</span><span class="o">=</span><span class="mi">1</span><span class="p">,</span> <span class="n">negative_slope</span><span class="o">=</span><span class="mf">0.2</span><span class="p">,</span> <span class="n">short_cut</span><span class="o">=</span><span class="kc