
<span class="sd">        hidden_dim (int): hidden dimension</span>
<span class="sd">        edge_input_dim (int): dimension of edge features</span>
<span class="sd">        num_layer (int, optional): number of hidden layers</span>
<span class="sd">        num_gru_layer (int, optional): number of GRU layers in each node update</span>
<span class="sd">        num_mlp_layer (int, optional): number of MLP layers in each message function</span>
<span class="sd">        num_s2s_step (int, optional): number of pr