torch_geometric.nn.conv.DirGNNConv

class DirGNNConv(conv: MessagePassing, alpha: float = 0.5, root_weight: bool = True)

Bases: Module

一个通用的包装器，用于在有向图上计算图卷积，如“边方向性改善异质图上的学习”论文中所述。 DirGNNConv将从源节点到目标节点以及从目标节点到源节点传递消息。

Parameters:

• conv (MessagePassing) – 使用的底层 MessagePassing 层。

• alpha (float, optional) – 用于加权入边和出边聚合的alpha系数，作为凸组合的一部分。 (默认值: 0.5)

• root_weight (bool, 可选) – 如果设置为 True，该层将把转换后的根节点特征添加到输出中。 (默认: True)

forward(x: Tensor, edge_index: Tensor) → Tensor

Return type:

Tensor

reset_parameters()

重置模块的所有可学习参数。