Struct BeamState 

pub struct BeamState {
    pub nodes: Vec<Vec<u32>>,
    pub scores: Vec<Vec<f64>>,
    pub tokens: Vec<Vec<Vec<u32>>>,
}

Live decoding state for a single batch of beam searches.

Shapes:

• nodes : [batch_size][beam_width] — current trie node per beam
• scores : [batch_size][beam_width] — accumulated log-probability
• tokens : [batch_size][beam_width][step] — partial SID decoded so far

Fields

nodes: Vec<Vec<u32>>

scores: Vec<Vec<f64>>

tokens: Vec<Vec<Vec<u32>>>

Implementations

impl BeamState

pub fn new(batch_size: usize, beam_width: usize) -> Self

Creates a blank state for batch_size queries, each with beam_width beams, all starting at the trie root (node 0) with log-prob 0.0.

pub fn batch_size(&self) -> usize

pub fn beam_width(&self) -> usize

pub fn step(&self) -> usize

Returns the current decoding step (number of tokens emitted so far). Assumes all beams in batch 0 are at the same step.

pub fn flat_nodes(&self) -> Vec<u32>

Flattens nodes into a single Vec<u32> of length batch_size * beam_width for bulk VNTK calls.

pub fn update_from_flat( &mut self, flat_nodes: &[u32], flat_scores: &[f64], flat_tokens: &[Vec<u32>], )

Replaces the state from a flat representation produced by the decoder. flat_nodes and flat_scores must have length batch_size * beam_width.

pub fn completed(&self, sid_length: usize) -> Vec<Vec<u32>>

Returns completed sequences (those whose tokens length == sid_length).

Trait Implementations

impl Clone for BeamState

fn clone(&self) -> BeamState

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for BeamState

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

Formats the value using the given formatter.

impl Display for BeamState

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

Formats the value using the given formatter.