Method get_nodes

graphlearn/python/graph.py:658–679

Get nodes from the graph. Args: node_type (string): Type of nodes, which should has been added by `Graph.node()`. ids (numpy.ndarray): ids of nodes. In sparse case, it must be 1D. offsets: (list): To get `SparseNodes`, whose dense shape is 2D, `offsets` indicat

(self, node_type, ids, offsets=None, shape=None)

    return self._edge_decoders

  def get_nodes(self, node_type, ids, offsets=None, shape=None):
    '''Get nodes from the graph.

    Args:
      node_type (string): Type of nodes, which should has been added by
        `Graph.node()`.
      ids (numpy.ndarray): ids of nodes. In sparse case, it must be 1D.
      offsets: (list): To get `SparseNodes`, whose dense shape is 2D,
        `offsets` indicates the number of nodes for each line.
        Default None means it is a dense `Nodes`.
      shape (tuple, Optional): Indicates the shape of nodes ids, attrs, etc.
        For dense case, default None means ids.shape. For sparse case, it
        must has a value which indicates the 2D dense shape.

    Return:
      A `Nodes` object or a `SparseNodes` object.
    '''
    if offsets is None:
      nodes = data.Nodes(ids, node_type, shape=shape, graph=self)
    else:
      nodes = data.SparseNodes(ids, offsets, shape, node_type, graph=self)
    return nodes

  def get_edges(self, edge_type, src_ids, dst_ids, edge_ids=None, offsets=None,
                shape=None, reverse=False):