首页 > 其他 > 详细

图中的节点和边

时间:2021-06-18 19:17:01      阅读:14      评论:0      收藏:0      [点我收藏+]

可以有节点特征,也可以有边特征。

节点特征,如node2vec,可以理解为[batch, feature]的dense矩阵

而边特征,可以理解为[batch, batch], (nnz =边的数量) 的sparse矩阵。sparse 中的values向量存的就是边特征的值。

另外,

对于加权图,用户可以将权重储存为一个边特征。

```

# 边 0->1, 0->2, 0->3, 1->3
>>> edges = th.tensor([0, 0, 0, 1]), th.tensor([1, 2, 3, 3])
>>> weights = th.tensor([0.1, 0.6, 0.9, 0.7])  # 每条边的权重
>>> g = dgl.graph(edges)
>>> g.edata[‘w‘] = weights  # 将其命名为 ‘w‘
>>> g
Graph(num_nodes=4, num_edges=4,
      ndata_schemes={}
      edata_schemes={‘w‘ : Scheme(shape=(,), dtype=torch.float32)})


参考:
https://docs.dgl.ai/en/0.6.x/guide_cn/graph-feature.html

图中的节点和边

原文:https://www.cnblogs.com/simpleminds/p/14900596.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!