dgl.DGLGraph.adj_tensors

DGLGraph.adj_tensors(fmt, etype=None)[source]

Return the adjacency matrix of edges of the given edge type as tensors of a sparse matrix representation. By default, a row of returned adjacency matrix represents the source of an edge and the column represents the destination. :param fmt: Either coo, csr or csc. :type fmt: str :param etype: The type names of the edges. The allowed type name formats are:

  • (str, str, str) for source node type, edge type and destination node type.

  • or one str edge type name if the name can uniquely identify a triplet format in the graph.

Can be omitted if the graph has only one type of edges.

Returns

If fmt is coo, returns a pair of source and destination node ID tensors. If fmt is csr or csc, return the CSR or CSC representation of the adjacency matrix as a triplet of tensors (indptr, indices, edge_ids). Namely edge_ids could be an empty tensor with 0 elements, in which case the edge IDs are consecutive integers starting from 0.

Return type

tuple[Tensor]

Examples

>>> g = dgl.graph(([0, 1, 2], [1, 2, 3]))
>>> g.adj_tensors('coo')
(tensor([0, 1, 2]), tensor([1, 2, 3]))
>>> g.adj_tensors('csr')
(tensor([0, 1, 2, 3, 3]), tensor([1, 2, 3]), tensor([0, 1, 2]))