dgl.function.e_mul_u

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

内置消息函数，通过对边和源节点特征进行元素级别的乘法运算来计算边上的消息，如果特征形状相同则直接计算；否则，首先广播特征到新的形状，然后执行元素级别运算。

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

参数：

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

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

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

示例

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