核心代码!!!
基于消息传递的模型有GCN 算法、GAT 算法、里面的核心就是Message Passing 消息传递机制。
'''
GCN参数补充解释
给出一个简化版本的 GCN 模型,帮助大家理解PGL框架实现消息传递的流程。
'''
##############################1################################
'''
'''
import paddle.fluid.layers as L
def gcn_layer(gw, feature, hidden_size, activation, name, norm=None):
"""
描述:通过GCN层计算新的节点表示
输入:gw - GraphWrapper对象
feature - 节点表示 (num_nodes, feature_size)
hidden_size - GCN层的隐藏层维度 int
activation - 激活函数 str
name - GCN层名称 str
norm - 标准化tensor float32 (num_nodes,),None表示不标准化
输出:新的节点表示 (num_nodes, hidden_size)
"""
# send函数
def send_func(src_feat, dst_feat, edge_feat):
"""
描述:用于send节点信