dgl.DGLGraph.all_edges

DGLGraph.all_edges(form='uv', sorted=False)

Return all the edges.

Parameters:

• form (str, optional) –

  The return form. Currently support:

  • ’all’ : a tuple (u, v, eid)
  • ’uv’ : a pair (u, v), default
  • ’eid’ : one eid tensor
• sorted (bool) – True if the returned edges are sorted by their src and dst ids.

Returns:

• A tuple of Tensors (u, v, eid) if form == ‘all’ – eid[i] is the ID of an edge between u[i] and v[i]. All edges are returned.
• A pair of Tensors (u, v) if form == ‘uv’ – An edge exists between u[i] and v[i]. If n edges exist between u and v, then u and v as a pair will appear n times.
• One Tensor if form == ‘eid’ – eid[i] is the ID of an edge in the graph.

Examples

The following example uses PyTorch backend.

>>> G = dgl.DGLGraph()
>>> G.add_nodes(3)
>>> G.add_edges([0, 0, 1], [1, 2, 2])   # (0, 1), (0, 2), (1, 2)
>>> G.all_edges()
(tensor([0, 0, 1]), tensor([1, 2, 2]))
>>> G.all_edges('all')
(tensor([0, 0, 1]), tensor([1, 2, 2]), tensor([0, 1, 2]))