Struct GraphMAEDecoder 

pub struct GraphMAEDecoder { /* private fields */ }

Decoder that reconstructs only masked node features (key efficiency gain).

Implementations

impl GraphMAEDecoder

pub fn new(hidden_dim: usize, output_dim: usize, num_layers: usize) -> Self

Create a decoder mapping hidden_dim -> output_dim.

pub fn decode( &self, latent: &[Vec<f32>], mask_idx: &[usize], re_mask: f32, ) -> Vec<Vec<f32>>

Decode latent for masked nodes. Applies re-masking (zeroing dims) for regularization.