dgl.function.u_mul_v¶

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

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

>>> import dgl