Skip to main content

MessageFunction

Trait MessageFunction 

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

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

Required Methods§

Source

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

计算消息

§Arguments
  • src_features - 源节点特征
  • edge_features - 边特征(可选)
  • dst_features - 目标节点特征
§Returns

返回计算得到的消息张量

Implementors§

Source§

impl MessageFunction<DenseTensor> for IdentityMessage

Available on crate feature tensor-gnn only.
Source§

impl MessageFunction<DenseTensor> for LinearMessage

Available on crate feature tensor-gnn only.