dgl.function.e_sub_v
- dgl.function.e_sub_v(lhs_field, rhs_field, out)
内建消息函数,通过对边上的特征进行计算来生成消息。如果特征形状相同,则执行 e 和 v 特征之间的元素级减法;否则,先将特征广播到新的形状,然后再执行元素级操作。
广播遵循 NumPy 语义。有关 NumPy 广播语义的更多详细信息,请参阅 https://docs.scipy.org.cn/doc/numpy/user/basics.broadcasting.html。
示例
>>> import dgl >>> message_func = dgl.function.e_sub_v('h', 'h', 'm')