torch_geometric.transforms.AddSelfLoops

class AddSelfLoops(attr: str = 'edge_weight', fill_value: Union[float, Tensor, str] = 1.0)

Bases: BaseTransform

向给定的同质或异质图添加自环 (函数名称: add_self_loops).

Parameters:

• attr (str, optional) – 传递给torch_geometric.utils.add_self_loops()的边权重或多维边特征的属性名称。 (默认: "edge_weight")

• fill_value (float 或 Tensor 或 str, 可选) – 生成自环边特征的方式（在 attr != None 的情况下）。 如果给定为 float 或 torch.Tensor，自环的边特征将直接由 fill_value 给出。 如果给定为 str，自环的边特征将通过聚合指向特定节点的所有边的特征来计算，根据一个归约操作。（"add", "mean", "min", "max", "mul"）。（默认值：1.）