dgl.function.copy_u

dgl.function.copy_u(u, out)

Builtin message function that computes message using source node feature.

Parameters:

• u (str) – The source feature field.

• out (str) – The output message field.

Examples

>>> import dgl
>>> message_func = dgl.function.copy_u('h', 'm')

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

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