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

参考：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）–更新功能。