   hidden_dims (list of int): hidden dimensions</span>
<span class="sd">        edge_input_dim (int, optional): dimension of edge features</span>
<span class="sd">        num_head (int, optional): number of attention heads</span>
<span class="sd">        negative_slope (float, optional): negative slope of leaky relu activation</span>
<span class="sd">        short_cut (bool, optional): use short cut or not</span>
<span class="sd">        batch_norm (bool, optional): apply batch normalization or not</span>
<