dgl.function.copy_edge

dgl.function.copy_edge(edge, out)[source]

Builtin message function that computes message using edge feature.

Parameters:
  • edge (str) – The edge feature field.
  • out (str) – The output message field.

Examples

>>> import dgl
>>> message_func = dgl.function.copy_edge(edge='h', out='m')

The above example is equivalent to the following user defined function:

>>> def message_func(edges):
>>>     return {'m': edges.data['h']}