dgl.DGLGraph.to_canonical_etype

DGLGraph.to_canonical_etype(etype)

将边类型转换为图中对应的规范边类型。

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

该函数期望给定的边类型名称能够唯一标识一个规范的边类型。如果不是这种情况，DGL将引发错误。

Parameters:

etype (str 或 (str, str, str)) – 如果 etype 是一个边类型 (str)，它返回图中对应的规范边类型。如果 etype 已经是一个规范边类型，它直接返回输入不变。

Returns:

与边类型对应的规范边类型。

Return type:

(str, str, str)

示例

以下示例使用PyTorch后端。

>>> import dgl
>>> import torch

创建一个异构图。

>>> g = dgl.heterograph({
...     ('user', 'follows', 'user'): ([0, 1], [1, 2]),
...     ('user', 'plays', 'game'): ([0, 1, 1, 2], [0, 0, 1, 1]),
...     ('developer', 'follows', 'game'): ([0, 1], [0, 1])
... })

将边类型映射到其对应的规范边类型。

>>> g.to_canonical_etype('plays')
('user', 'plays', 'game')
>>> g.to_canonical_etype(('user', 'plays', 'game'))
('user', 'plays', 'game')

另请参阅

canonical_etypes