对称边权重

class SymmetricEdgeWeighting(**kwargs)

基础：EdgeWeighting

通过入度和出度的逆平方根的乘积来规范化消息。

初始化模块。

Parameters:

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

方法总结

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

计算边的权重。

方法文档

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