dgl.DGLGraph.register_apply_edge_func

DGLGraph.register_apply_edge_func(func)

Register global edge apply function.

Once registered, func will be used as the default apply edge function in apply_edges().

Parameters:func (callable) – Apply function on the edge. The function should be an Edge UDF.

See also

apply_edges(), register_apply_node_func()