dgl.function.u_dot_e

dgl.function.u_dot_e(lhs_field, rhs_field, out)

Builtin message function that computes message by performing binary operation dot between src feature and edge feature.

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

Examples

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