torch_geometric.transforms.AddRandomWalkPE
- class AddRandomWalkPE(walk_length: int, attr_name: Optional[str] = 'random_walk_pe')[source]
Bases:
BaseTransform
Adds the random walk positional encoding from the “Graph Neural Networks with Learnable Structural and Positional Representations” paper to the given graph (functional name:
add_random_walk_pe
).