DGL官方教程--API--dgl.udf

最新推荐文章于 2024-12-08 02:00:24 发布

翻译最新推荐文章于 2024-12-08 02:00:24 发布 · 562 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：https://docs.dgl.ai/en/latest/generated/dgl.propagate.prop_edges_dfs.html

文章标签：

#pytorch

DGL库笔记专栏收录该内容

13 篇文章

订阅专栏

本文介绍DGL图神经网络库中消息传播模块的详细功能，包括如何通过不同的遍历算法如BFS、DFS和拓扑排序进行消息传播，以及如何在节点和边上传播消息。介绍了prop_nodes和prop_edges函数的参数和使用方法。

部署运行你感兴趣的模型镜像

参考：https://docs.dgl.ai/en/latest/api/python/propagate.html

dgl.propagate

消息传播模块。

按照图遍历顺序传播消息并执行计算。

prop_nodes_XXX 调用遍历算法XXX并在每次迭代时在访问节点集上触发pull()。

prop_edges_YYY应用遍历算法YYY并在每次迭代时在访问边缘集上触发send_and_recv() 。

`prop_nodes`(graph, nodes_generator[, …])	`dgl.DGLGraph.prop_nodes()`的功能方法。
`prop_edges`(graph, edges_generator[, …])	`dgl.DGLGraph.prop_edges()`的功能方法。
`prop_nodes_bfs`(graph, source[, reverse, …])	对BFS生成的节点边界进行消息传播。
`prop_nodes_topo`(graph[, reverse, …])	对拓扑遍历生成的节点边界进行消息传播。
`prop_edges_dfs`(graph, source[, reverse, …])	对DFS生成的边缘边界进行消息传播。

dgl.propagate.prop_nodes

dgl.propagate.prop_nodes(graph,nodes_generator,message_func ='default',reduce_func ='default',apply_node_func ='default')

dgl.DGLGraph.prop_nodes()的功能方法。

参数：	node_generators（generator）–节点边界的生成器。 message_func（callable ，optional）–消息功能。 reduce_func（callable ，optional）– reduce函数。 apply_node_func（callable ，optional）–更新功能。

dgl.propagate.prop_edges

dgl.propagate.prop_edges(graph,Edges_generator,message_func ='default',reduce_func ='default',apply_node_func ='default')

dgl.DGLGraph.prop_edges()的功能方法。

参数：	edges_generator（generator）–边缘边界的生成器。 message_func（callable ，optional）–消息功能。 reduce_func（callable ，optional）– reduce函数。 apply_node_func（callable ，optional）–更新功能。

dgl.propagate.prop_nodes_bfs

dgl.propagate.prop_nodes_bfs(graph,source,reverse = False,message_func ='default',reduce_func ='default',apply_node_func ='default')

对BFS生成的节点边界进行消息传播。

参数：	graph（DGLGraph）–图形对象。 source (list, tensor of nodes) –源节点。 reverse (bool, optional) –如果为true，则沿边缘方向移动。 message_func（callable ，optional）–消息功能。 reduce_func（callable ，optional）– reduce函数。 apply_node_func（callable ，optional）–更新功能。

dgl.propagate.prop_nodes_topo

dgl.propagate.prop_nodes_topo(graph,reverse = False,message_func ='default',reduce_func ='default',apply_node_func ='default' )

使用拓扑顺序生成的节点边界进行消息传播。

参数：	graph（DGLGraph）–图形对象。 reverse (bool, optional)–如果为true，则沿边缘方向移动。 message_func（callable ，optional）–消息功能。 reduce_func（callable ，optional）– reduce函数。 apply_node_func（callable ，optional）–更新功能。

dgl.propagate.prop_edges_dfs

dgl.propagate.prop_edges_dfs(graph,source,reverse = False,has_reverse_edge = False,has_nontree_edge = False,message_func ='default',reduce_func ='default',apply_node_func ='default')

使用标记的DFS生成的边缘边界进行消息传播。

参数：	graph（DGLGraph）–图形对象。 source (list, tensor of nodes)–源节点。 reverse (bool, optional) –如果为true，则沿边缘方向移动。 message_func（callable ，optional）–消息功能。 reduce_func（callable ，optional）– reduce函数。 apply_node_func（callable ，optional）–更新功能。