dgl.function.v_sub_uο
- dgl.function.v_sub_u(lhs_field, rhs_field, out)ο
Builtin message function that computes a message on an edge by performing element-wise sub between features of v and u 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:
Examples
>>> import dgl >>> message_func = dgl.function.v_sub_u('h', 'h', 'm')