torch_geometric.transforms.ToUndirected

class ToUndirected(reduce: str = 'add', merge: bool = True)[source]

Bases: BaseTransform

将同质或异质图转换为无向图，使得对于每一条边 \((i,j) \in \mathcal{E}\)，都有 \((j,i) \in \mathcal{E}\)（函数名称：to_undirected）。在异质图中，将为所有现有的边类型添加“反向”连接。

Parameters:

  • reduce (str, optional) – 用于合并边缘特征的reduce操作 ("add", "mean", "min", "max", "mul"). (默认: "add")

  • merge (bool, 可选) – 如果设置为 False，将为指向相同源节点和目标节点类型的连接创建反向边类型。 如果设置为 True，反向边将被合并到原始关系中。 此选项仅在 HeteroData 图数据中有效。 (默认: True)