parameters: (self, edge_index, node_vec, **kwargs)
docstring: Args:
        edge_index: A Tensor. Edge index of subgraph.
        node_vec: A Tensor. node feature embeddings with shape
        [batchsize, dim].
      Returns:
        A tensor. output embedding with shape [batch_size, output_dim].
docstring_summary: Args:
        edge_index: A Tensor. Edge index of subgraph.
        node_vec: A Tensor. node feature embeddings with shape
        [batchsize, dim].
      Returns:
        A tensor. output embedding with shape [batch_size, output_dim].
docstring_tokens: Args : edge_index : A Tensor . Edge index of subgraph . node_vec : A Tensor . node feature embeddings with shape [ batchsize dim ] . Returns : A tensor . output embedding with shape [ batch_size output_dim ] .
function: def forward(self, edge_index, node_vec, **kwargs):
      """
      Args:
        edge_index: A Tensor. Edge index of subgraph.
        node_vec: A Tensor. node feature embeddings with shape
        [batchsize, dim].
      Returns:
        A tensor. output embedding with shape [batch_size, output_dim].
      """
      pass
function_tokens: def forward ( self , edge_index , node_vec , * * kwargs ) : pass