dgl.function.e_add_v

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

内置消息函数,通过对边 (e) 特征和节点 (v) 特征执行元素wise 加法来计算边上的消息,如果特征形状相同;否则,它首先将特征广播到新的形状,然后执行元素wise 操作。

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

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

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

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

示例

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