您当前的位置: 首页 > 

静静喜欢大白

暂无认证

  • 3浏览

    0关注

    521博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

百度7天GNN学习-实战2理解(MPNN中send和recv白话理解)

静静喜欢大白 发布时间:2021-01-13 14:08:02 ,浏览量:3

核心代码!!!

基于消息传递的模型有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节点信
关注
打赏
1510642601
查看更多评论
立即登录/注册

微信扫码登录

0.0364s