【dgl学习】dgl处理图中的节点/边的属性/特征/类型

原创

已于 2022-08-24 20:13:33 修改 · 3.4k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#学习 #深度学习 #python

于 2022-08-23 10:14:19 首次发布

本文介绍了DGL库中用于处理图数据的几个关键函数，包括DGLGraph.nodes用于获取或设置节点特征，DGLGraph.edges用于访问边数据，以及DGLGraph.edata用于操作边特征。通过示例展示了在同构图和异构图上如何操作节点和边的ID及特征，对于理解和使用DGL进行图神经网络编程十分有帮助。

1 dgl.DGLGraph.nodes

函数的主要功能有两个

返回图中单个节点类型的ID
获取或者设置单个节点类型的特征

主要有两个应用同构图和异构图：

1.1 举个例子

import dgl
import torch

# 定义同构图
g = dgl.graph((torch.tensor([0, 1]), torch.tensor([1, 2])))
# 定义异构图
hg = dgl.heterograph({
   
   
    ('user', 'follows', 'user'): (torch.tensor([0, 1]), torch.tensor([1, 2])),
    ('user', 'plays', 'game'): (torch.tensor([3,