图神经网络(Graph Neural Networks,GNN)是一种用于处理图数据的深度学习模型。PyG(PyTorch Geometric)是一个基于PyTorch的图神经网络库,提供了丰富的图神经网络算子和功能。本文将介绍PyG中常用的图神经网络算子,并附上相应的源代码示例。
- 图的表示
在PyG中,使用torch_geometric.data.Data
对象来表示图数据。该对象包含两个属性:x
表示节点特征矩阵,edge_index
表示边的索引。下面是一个创建图数据的示例:
import torch
from torch_geometric.data import Data
x = torch.tensor([