Trait MessageFunction 

pub trait MessageFunction<H: TensorBase>: Send + Sync {
    // Required method
    fn message(
        &self,
        src_features: &H,
        edge_features: Option<&H>,
        dst_features: &H,
    ) -> H;
}

消息函数 trait：定义边上的消息计算

Required Methods

fn message( &self, src_features: &H, edge_features: Option<&H>, dst_features: &H, ) -> H

计算消息

Arguments

• src_features - 源节点特征
• edge_features - 边特征（可选）
• dst_features - 目标节点特征

Returns

返回计算得到的消息张量

Implementors

impl MessageFunction<DenseTensor> for IdentityMessage

Available on crate feature tensor-gnn only.

impl MessageFunction<DenseTensor> for LinearMessage

Available on crate feature tensor-gnn only.