dgl.DGLGraph.canonical_etypes

property DGLGraph.canonical_etypes

返回图中的所有规范边类型。

规范边类型是一个字符串三元组 (str, str, str)，表示源节点类型、边类型和目标节点类型。

返回值:

列表形式的所有规范边类型三元组。

返回类型:

list[(str, str, str)]

备注

DGL 在内部为每种边类型分配一个整数 ID。返回的边类型名称按其 ID 排序。

另请参阅

etypes

示例

以下示例使用 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]))
... })
>>> g.canonical_etypes
[('user', 'follows', 'user'),
 ('user', 'follows', 'game'),
 ('user', 'plays', 'game')]