dgl.function.u_mul_e

dgl.function.u_mul_e(lhs_field, rhs_field, out)

Builtin message function that computes a message on an edge by performing element-wise mul between features of u and e if the features have the same shape; otherwise, it first broadcasts the features to a new shape and performs the element-wise operation.

Broadcasting follows NumPy semantics. Please see https://docs.scipy.org/doc/numpy/user/basics.broadcasting.html for more details about the NumPy broadcasting semantics.

Parameters:

  • lhs_field (str) – The feature field of u.

  • rhs_field (str) – The feature field of e.

  • out (str) – The output message field.

Examples

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