【dgl学习】dgl.canonical_etypes函数解析

一穷二白到年薪百万

于 2022-08-23 09:01:55 发布

阅读量790

点赞数

CC 4.0 BY-SA版权

分类专栏： # dgl 文章标签：深度学习

本文链接：https://blog.youkuaiyun.com/zfhsfdhdfajhsr/article/details/126477556

dgl 专栏收录该内容

11 篇文章

订阅专栏

该博客介绍了如何使用DGL库从异构图中获取规范化的边类型。`canonical_etypes`函数返回图中所有独特的三元组形式（源节点类型，边类型，目标节点类型），例如`('user','follows','user')`。示例展示了创建包含不同边类型的图，并调用`canonical_etypes`来列举所有边类型。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

函数功能：以三元组的形式返回图中定义的边。
下面是官方给出的解释

Return all the canonical edge types in the graph.

A canonical edge type is a string triplet (str, str, str) for source node type, edge type and destination node type.

Returns
All the canonical edge type triplets in a list.

Return type
list[(str, str, str)]

举个例子

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')]