dgl.contrib.graph_store.SharedMemoryDGLGraph.push¶
-
SharedMemoryDGLGraph.
push
(u, message_func='default', reduce_func='default', apply_node_func='default', inplace=True)[source]¶ Send message from the node(s) to their successors and update them.
Optionally, apply a function to update the node features after receive.
In the graph store, all updates are written inplace.
- Parameters
u (int, iterable of int, or tensor) – The node(s) to push messages out.
message_func (callable, optional) – Message function on the edges. The function should be an
Edge UDF
.reduce_func (callable, optional) – Reduce function on the node. The function should be a
Node UDF
.apply_node_func (callable, optional) – Apply function on the nodes. The function should be a
Node UDF
.inplace (bool, optional) – The value is always True.