dgl.function.e_mul_v

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

内置消息函数，通过在边上执行元素乘法计算消息。如果边 (e) 和顶点 (v) 的特征形状相同，则直接执行元素乘法；否则，先将特征广播到新的形状，再执行元素级操作。

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

参数:

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

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

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

示例

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