dgl.function.u_mul_v

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

Builtin message function that computes message by performing binary operation mul between src feature and dst feature.

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

Examples

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