![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
图神经网络
Henry Fox
这个作者很懒,什么都没留下…
展开
-
pytorch加载参数设备选择 2021-12-12
https://blog.csdn.net/bc521bc/article/details/85623515 由于pytorch的模型保存的时候记录了device信息,所以加载的时候需要手动指定设备。 通过:, map_location={'cuda:1':'cuda:0'}的写法转载 2021-12-12 14:15:43 · 164 阅读 · 0 评论 -
GNN包torch_geometric使用 2021-09-02
torch_geometric包是一个进行GNN实现的非常方便的包,官网是 https://pytorch-geometric.readthedocs.io/en/latest/notes/introduction.html 数据集格式 使用的数据集的格式如下例所示: >>> import torch >>> from torch_geometric.data import Data >>> edge_index = torch.tensor([[0,原创 2021-09-02 16:13:48 · 2048 阅读 · 5 评论 -
神经网络-GAT 2021-07-20
每一个节点分别对应于一个hhh,WWW是这里的learnable weight matrix,负责将每个hjh_jhj进行转换,变为WhjWh_jWhj。 利用attention机制,αij=e( LeakyReLU (waT‾[Whi∥Whj]))∑k∈N(i)e( LeakyReLU (waT‾[Whi∥Whj]))\alpha_{i j}=\frac{e^{ \left(\text { LeakyReLU }\left(\overline{w_{a}^{T}原创 2021-07-20 16:35:33 · 2211 阅读 · 0 评论