Struct RandomWalkPe 

pub struct RandomWalkPe {
    pub walk_length: usize,
    pub projection: Array2<f64>,
    pub pe_dim: usize,
}

Random Walk Positional Encoding (RWPE).

Computes the diagonal of P^k for k=1..K where P is the random walk transition matrix. The landing probabilities R_ii = (P^k)_ii encode local structural information around each node.

Fields

walk_length: usize

Number of random walk steps (K)

projection: Array2<f64>

Linear projection: [walk_length, pe_dim]

pe_dim: usize

Output PE dimension

Implementations

impl RandomWalkPe

pub fn new(walk_length: usize, pe_dim: usize) -> Self

Create a new RWPE module.

Arguments

• walk_length - Number of random walk steps K
• pe_dim - Output dimension for the positional encoding

pub fn compute_landing_probs(&self, adj: &CsrMatrix) -> Array2<f64>

Compute random walk landing probabilities.

Returns [n, walk_length] where entry (i, k) = (P^{k+1})_{ii}.

pub fn forward(&self, adj: &CsrMatrix) -> Array2<f64>

Compute RWPE and project to pe_dim.

Arguments

• adj - Adjacency matrix

Returns

Positional encoding [n, pe_dim]

Trait Implementations

impl Clone for RandomWalkPe

fn clone(&self) -> RandomWalkPe

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for RandomWalkPe

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.