dgl.function.e_sub_u

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

内置消息函数，用于通过对边上的特征 e 和 u 执行逐元素减法来计算消息，前提是特征具有相同的形状；否则，它会先将特征广播到新形状再执行逐元素操作。

广播遵循 NumPy 语义。请参阅 https://docs.scipy.org.cn/doc/numpy/user/basics.broadcasting.html 以了解有关 NumPy 广播语义的更多详情。

参数:

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

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

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

示例

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