torch_geometric.transforms.AddRandomWalkPE

class AddRandomWalkPE(walk_length: int, attr_name: Optional[str] = 'random_walk_pe')

Bases: BaseTransform

将来自“具有可学习结构和位置表示的图神经网络”论文的随机游走位置编码添加到给定的图中（功能名称：add_random_walk_pe）。

Parameters:

• walk_length (int) – 随机游走的步数。

• attr_name (str, optional) – 要添加位置编码的数据对象的属性名称。如果设置为 None，将会被连接到 data.x。 (默认值: "random_walk_pe")