dgl.function.v_sub_u

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

内置消息函数，通过对v和u的特征进行元素级别减法（如果特征形状相同）来计算边上的消息；否则，它首先将特征广播到新的形状，然后执行元素级别操作。

广播遵循NumPy语义。有关NumPy广播语义的更多详细信息，请参阅 https://docs.scipy.org.cn/doc/numpy/user/basics.broadcasting.html。

参数：

• lhs_field (str) – v的特征字段。

• rhs_field (str) – u的特征字段。

• out (str) – 输出消息字段。

示例

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