ToLevi

class dgl.transforms.ToLevi

基类: BaseTransform

此函数将原始图转换为其异构 Levi 图，通过将边转换为中间节点实现，仅支持同构有向图。

示例

>>> import dgl
>>> import torch as th
>>> from dgl import ToLevi

>>> transform = ToLevi()
>>> g = dgl.graph(([0, 1, 2, 3], [1, 2, 3, 0]))
>>> g.ndata['h'] = th.randn((g.num_nodes(), 2))
>>> g.edata['w'] = th.randn((g.num_edges(), 2))
>>> lg = transform(g)
>>> lg
Grpah(num_nodes={'edge': 4, 'node': 4},
      num_edges={('edge', 'e2n', 'node'): 4,
                 ('node', 'n2e', 'edge'): 4},
      metagraph=[('edge', 'node', 'e2n'),
                 ('node', 'edge', 'n2e')])
>>> lg.nodes('node')
tensor([0, 1, 2, 3])
>>> lg.nodes('edge')
tensor([0, 1, 2, 3])
>>> lg.nodes['node'].data['h'].shape
torch.Size([4, 2])
>>> lg.nodes['edge'].data['w'].shape
torch.Size([4, 2])