Struct GATEncoder 

pub struct GATEncoder { /* private fields */ }

Multi-layer GAT encoder for GraphMAE.

Implementations

impl GATEncoder

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

Build an encoder with num_layers GAT layers.

pub fn encode(&self, features: &[Vec<f32>], adj: &[Vec<usize>]) -> Vec<Vec<f32>>

Encode node features through all GAT layers.