Struct GNNEncoder 

pub struct GNNEncoder { /* private fields */ }

GNN Encoder for syndrome graphs

Implementations

impl GNNEncoder

pub fn new(config: GNNConfig) -> Result<Self>

Create a new GNN encoder

Errors

Returns an error if embed_dim is not divisible by num_heads.

pub fn encode(&self, graph: &DetectorGraph) -> Result<Array2<f32>>

Encode a detector graph

pub fn get_intermediate_embeddings( &self, graph: &DetectorGraph, layer_idx: usize, ) -> Result<Vec<Vec<f32>>>

Get node embeddings without output projection (for debugging)

pub fn config(&self) -> &GNNConfig

Get the configuration

Trait Implementations

impl Clone for GNNEncoder

fn clone(&self) -> GNNEncoder

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GNNEncoder

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

Formats the value using the given formatter.