文章目录 前言 一、导入相关库 二、加载Cora数据集 三、定义GAT网络 3.1 定义GATConv层 3.1.1 消息传递阶段(message) 3.1.2 消息聚合阶段(aggregate) 3.1.3 节点更新阶段(update) 3.1.4 定义传播过程(propagate) 3.1.5 定义GATConv层 3.2 定义GAT网络 四、定义模型 五、模型训练