dgl.to_homo

dgl.to_homo(G)[source]

Convert the given heterogeneous graph to a homogeneous graph.

The returned graph has only one type of nodes and edges.

Node and edge types are stored as features in the returned graph. Each feature is an integer representing the type id, which can be used to retrieve the type names stored in G.ntypes and G.etypes arguments.

Parameters:G (DGLHeteroGraph) – Input heterogeneous graph.
Returns:A homogeneous graph. The parent node and edge type/ID are stored in columns dgl.NTYPE/dgl.NID and dgl.ETYPE/dgl.EID respectively.
Return type:DGLHeteroGraph

Examples

>>> follows_g = dgl.graph([(0, 1), (1, 2)], 'user', 'follows')
>>> devs_g = dgl.bipartite([(0, 0), (1, 1)], 'developer', 'develops', 'game')
>>> hetero_g = dgl.hetero_from_relations([follows_g, devs_g])
>>> homo_g = dgl.to_homo(hetero_g)
>>> homo_g.ndata
{'_TYPE': tensor([0, 0, 0, 1, 1, 2, 2]), '_ID': tensor([0, 1, 2, 0, 1, 0, 1])}
First three nodes for 'user', next two for 'developer' and the last two for 'game'
>>> homo_g.edata
{'_TYPE': tensor([0, 0, 1, 1]), '_ID': tensor([0, 1, 0, 1])}
First two edges for 'follows', next two for 'develops'

See also

dgl.to_hetero()