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.