dgl.function.e_div_v
- dgl.function.e_div_v(lhs_field, rhs_field, out)
内置消息函数,通过对边 (e) 和节点 (v) 的特征执行逐元素除法来计算边上的消息。如果特征形状相同则直接计算;否则,它首先将特征广播到新的形状,然后执行逐元素操作。
广播遵循 NumPy 的语义。请参阅 https://docs.scipy.org.cn/doc/numpy/user/basics.broadcasting.html 以获取有关 NumPy 广播语义的更多详细信息。
示例
>>> import dgl >>> message_func = dgl.function.e_div_v('h', 'h', 'm')