Method E

graphlearn/python/graph.py:586–618

Entry of GSL, starting from EDGE. Args: edge_type (string): The type of edge which is the entry of query. feed (None| (np.ndarray, np.ndarray) | types.GeneratorType | `Edges`): None: Default. Sample edges with the following .shuffle and .batch API. (np.ndarray, np

(self,
        edge_type,
        feed=None,
        reverse=False)

    return source_node

  def E(self,
        edge_type,
        feed=None,
        reverse=False):
    """ Entry of GSL, starting from EDGE.

    Args:
      edge_type (string): The type of edge which is the entry of query.
      feed (None| (np.ndarray, np.ndarray) | types.GeneratorType | `Edges`):
        None: Default. Sample edges with the following .shuffle and .batch API.
          (np.ndarray, np.ndarray): src_ids, dst_ids. Get edges of the given
          (src_ids, dst_ids) and given edge_type. src_ids and dst_ids must be
          the same shape, dtype is int.
        types.Generator: A generator of (numpy.ndarray, numpy.ndarray). Get
          edges of generated (src_ids, dst_ids) and given edge_type.
        `Edges`: An `Edges` object.
    """
    if feed is not None:
      raise NotImplementedError("`feed` is not supported for now.")
    dag = gsl.Dag(self)
    if reverse:
      edge_type = edge_type + "_reverse"

    dag = gsl.Dag(self)
    params = {pywrap.kEdgeType: edge_type}
    source_node = gsl.TraverseSourceEdgeDagNode(
      dag, op_name="GetEdges", params=params)
    source_node.set_output_field(pywrap.kEdgeIds)
    source_node.set_path(edge_type, pywrap.NodeFrom.NODE)

    # Add sink node to dag
    gsl.SinkNode(dag)
    return source_node

  def get_topology(self):
    """ Get the topology of the graph.