Struct BeamStepInput 

pub struct BeamStepInput {
    pub log_probs: Vec<Vec<f64>>,
}

Log-probabilities (or raw logits) for each beam at one decoding step.

Shape: [beam_width][vocab_size] — each row is the distribution for one beam.

Fields

log_probs: Vec<Vec<f64>>

log_probs[beam_i][vocab_j] = log P(token j | history of beam i).

Implementations

impl BeamStepInput

pub fn new(log_probs: Vec<Vec<f64>>) -> Self

Construct directly from pre-computed log-probabilities.

pub fn from_logits(logits: Vec<Vec<f64>>, temperature: f64) -> Self

Construct from raw logits: apply temperature scaling then log-softmax.

pub fn num_beams(&self) -> usize

Number of beams (rows) in this input.

pub fn vocab_size(&self) -> usize

Vocabulary size inferred from the first row (0 if empty).