边权重

class EdgeWeighting(**kwargs)

基础类: Module

边权重的基础类。

初始化模块。

Parameters:

kwargs – 忽略的基于关键字的参数。

属性摘要

needs_message

边缘加权是否需要访问消息

方法总结

forward(source, target[, message, x_e])

计算边的权重。

属性文档

needs_message: ClassVar[bool] = False

边缘加权是否需要访问消息

方法文档

abstract forward(source: Tensor, target: Tensor, message: Tensor | None = None, x_e: Tensor | None = None) → Tensor

计算边的权重。

Parameters:

• source (Tensor) – 形状: (num_edges,) 源索引。

• target (Tensor) – 形状: (num_edges,) 目标索引。

• 消息 (Tensor | None) – 形状 (num_edges, dim) 实际要加权的消息

• x_e (Tensor | None) – 形状 (num_nodes, dim) 节点状态直到加权点

Returns:

形状: (num_edges, dim) 使用边权重加权的消息。

Return type:

Tensor