dgl.prop_nodes_bfs

dgl.prop_nodes_bfs(graph, source, message_func, reduce_func, reverse=False, apply_node_func=None)[source]

使用BFS生成的节点前沿进行消息传播。

Parameters:

  • graph (DGLGraph) – The graph object.

  • source (list, tensor of nodes) – Source nodes.

  • message_func (callable) – The message function.

  • reduce_func (callable) – The reduce function.

  • reverse (bool, optional) – If true, traverse following the in-edge direction.

  • apply_node_func (callable, optional) – The update function.

另请参阅

dgl.traversal.bfs_nodes_generator