dgl.DGLGraph.metagraph

DGLGraph.metagraph()[source]

返回异构图的元图。

异构网络的元图(或网络 schema)指定了节点集以及节点之间边的类型的约束。有关正式定义,请参阅 Yizhou et al.

返回值:

元图。

返回类型:

networkx.MultiDiGraph

示例

以下示例使用 PyTorch 后端。

>>> import dgl
>>> import torch
>>> g = dgl.heterograph({
...     ('user', 'follows', 'user'): (torch.tensor([0, 1]), torch.tensor([1, 2])),
...     ('user', 'follows', 'game'): (torch.tensor([0, 1, 2]), torch.tensor([1, 2, 3])),
...     ('user', 'plays', 'game'): (torch.tensor([1, 3]), torch.tensor([2, 3]))
... })
>>> meta_g = g.metagraph()
>>> meta_g.nodes()
NodeView(('user', 'game'))
>>> meta_g.edges()
OutMultiEdgeDataView([('user', 'user'), ('user', 'game'), ('user', 'game')])