![](https://img-blog.csdnimg.cn/20201014180756926.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
图神经网络
qq_1206630874
学生
展开
-
异构图如何使用attention
@[图神经网络](dgl 异构图,如果加入attention)本次测试使用两种类型节点 句子节点,单词节点多头注意力参考这里import torchimport dglimport dgl.function as fnimport torch.nn as nnimport torch.nn.functional as Fdef generate_empty_graph(num): empty_graphs=[] for i in range(num): g原创 2021-10-18 19:08:18 · 167 阅读 · 0 评论 -
图神经网络库:dgl,分割子图问题
将一个图划分为多个子图,使用dgl.node_subgraph()分割的子图会多出 edges_schemes:{“_ID” …}举例:import dglimport torchsrc=[0,1,2,3,4,5]dst=[1,2,3,0,5,4]feats=torch.tensor([[0,0,0],[1,1,1],[2,2,2],[3,3,3],[4,4,4],[5,5,5]])g=dgl.graph((torch.tensor(src),torch.tensor(dst)))g.原创 2021-09-10 21:18:54 · 1231 阅读 · 0 评论 -
dgl batch unbatch
@图神经网络工具dgl两层dgl.batch的图进行unbatch操作将会分开所有的图,如果想保留第一次batch的图怎么操作bg=dgl.batch([g1,g2,g3])new_graph=dgl.graph(bg.edges())new_graph.ndata["feats"]=bg.ndata["feats"]graphs.append(new_graph)...原创 2021-08-29 16:52:58 · 510 阅读 · 1 评论