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