Struct GraphNode 

pub struct GraphNode {
    pub node_type: GraphNodeType,
    pub id: usize,
    pub layer: usize,
    pub position: usize,
    pub token_embedding: Option<TokenEmbeddingNode>,
    pub hidden_state: Option<HiddenStateNode>,
    pub attention_output: Option<AttentionOutputNode>,
    pub ffn_output: Option<FFNOutputNode>,
}

Graph node wrapper

Fields

node_type: GraphNodeType

Node type

id: usize

Unique node ID

layer: usize

Layer number (for layer-specific nodes)

position: usize

Position in sequence

token_embedding: Option<TokenEmbeddingNode>

Optional token embedding data

hidden_state: Option<HiddenStateNode>

Optional hidden state data

attention_output: Option<AttentionOutputNode>

Optional attention output data

ffn_output: Option<FFNOutputNode>

Optional FFN output data

Implementations

impl GraphNode

pub fn token_embedding( id: usize, token_id: usize, position: usize, embedding: DenseTensor, ) -> Self

Create a token embedding node

pub fn hidden_state( id: usize, layer: usize, position: usize, state: DenseTensor, ) -> Self

Create a hidden state node

pub fn attention_output( id: usize, layer: usize, head: usize, query_pos: usize, attended_positions: Vec<usize>, weights: Vec<f64>, output: DenseTensor, ) -> Self

Create an attention output node

pub fn ffn_output( id: usize, layer: usize, position: usize, output: DenseTensor, ) -> Self

Create a FFN output node

pub fn get_embedding(&self) -> Option<&TokenEmbeddingNode>

Get the embedding if this is a token embedding node

pub fn get_hidden_state(&self) -> Option<&HiddenStateNode>

Get the hidden state if this is a hidden state node

pub fn get_attention_output(&self) -> Option<&AttentionOutputNode>

Get the attention output if this is an attention output node

pub fn get_ffn_output(&self) -> Option<&FFNOutputNode>

Get the FFN output if this is a FFN output node

Trait Implementations

impl Clone for GraphNode

fn clone(&self) -> GraphNode

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for GraphNode

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

Formats the value using the given formatter.