
图神经网络
文章平均质量分 95
只是一个代号不必认真
这个作者很懒,什么都没留下…
展开
-
torch_geometric实现GCN和LightGCN
使用torch_geometric实现GCN和LightGCN,以后可能要用,做一下备份。原创 2023-08-15 18:35:05 · 605 阅读 · 0 评论 -
GCN torch_geometric utils scatter_方法源码解析及样例
scatter_方法源码解析源码位置源码及注释torch_scatter方法scatter_方法解析常规运行节点特征维度dim_size(运算后返回的维度)参数index(索引)参数name(聚合方式)感谢及参考博文源码位置torch_geometric\utils\scatter.py源码及注释代码相当于在原有torch_scatter包的基础上进行二次开发。import torch_scatterdef scatter_(name, src, index, dim=0, dim_size原创 2021-05-15 17:49:43 · 1866 阅读 · 2 评论 -
GCN torch_geometric message_passing方法源码解析
message_passing方法源码解析前言源码位置源码及注释MessagePassing类propagate方法感谢及参考博文前言最近在看关于图神经网络相关论文,涉及到论文复现,所以要啃源码。网上虽然有一部分对源码的解析,如torch_geometric()方法,但是在深入到里面具体方法,资料寥寥无几。特将自己学习笔记分享,一来方便自己以后查阅,二来给和我一样苦难的同胞一点点思路,三来方便热心网友补充,最后为后面大神分享代码或思路抛砖引玉。^_^源码位置torch_geometric\nn\原创 2021-05-15 16:19:52 · 1630 阅读 · 0 评论