DGL GAT

最新推荐文章于 2023-12-05 19:22:06 发布

KrystalKarlieKarina

最新推荐文章于 2023-12-05 19:22:06 发布

阅读量1.7k

点赞数 3

分类专栏：论文学习

本文链接：https://blog.csdn.net/lovekkarlie_/article/details/117922366

版权

这篇博客探讨了在DGL库中实现GAT（Graph Attention Network）时遇到的问题及解决方案，包括graph数据类型错误，特别是对于DGLHeteroGraph的要求，以及在GPU上运行DGLGraph时遇到的设备不匹配问题。博主分享了如何将graph移动到GPU以及避免多卡环境中的错误。

摘要由CSDN通过智能技术生成

GAT
DGL document
DGL GAT
DGL官方教程GAT

# Case 1: Homogeneous graph
pip install dgl

import dgl
import numpy as np
import torch as th
from dgl.nn import GATConv

g = dgl.graph(([0,1,2,3,2,5], [1,2,3,4,0,3]))
# print(g)
# print(g.nodes())
# pri