论文+代码复现
行走天涯的豆沙包
人生天地间,忽如远行客。
展开
-
GAT源码剖析
题解:首先挂出核心公式和训练过程生成的aij注意系数layer.pyimport numpy as npimport torchimport torch.nn as nnimport torch.nn.functional as Fclass GraphAttentionLayer(nn.Module): """ Simple GAT layer, similar to https://arxiv.org/abs/1710.10903 """ def __原创 2021-01-06 17:29:44 · 1894 阅读 · 3 评论 -
GCN——源码剖析
图学习layer.pyimport mathimport torchfrom torch.nn.parameter import Parameter # 可以用parameter()函数from torch.nn.modules.module import Module # 定义网络层的模块'''parameter()将一个不可训练的类型Tensor转换成可以训练的类型parameter并将其绑定到这个module里面,所以经过类型转换这个就变成了模型的一部分,成为了模型原创 2021-01-05 17:23:34 · 1201 阅读 · 2 评论