Struct BeamSearchDecoder 

pub struct BeamSearchDecoder { /* private fields */ }

Beam search decoder for Whisper

Implementations

impl BeamSearchDecoder

pub fn new(config: DecodingConfig) -> Self

Create a new beam search decoder

pub fn decode( &self, token_probs: &[Vec<f32>], initial_token: u32, vocab_size: usize, eos_token: u32, _pad_token: u32, ) -> Result<Vec<u32>>

Decode token probabilities to a sequence using beam search

Arguments

• token_probs - Matrix of shape (seq_len, vocab_size) with log probabilities per token
• initial_token - Starting token (usually language token or BOS)
• vocab_size - Size of vocabulary
• eos_token - End-of-sequence token ID
• _pad_token - Padding token ID (reserved for future use)

Returns

Vector of token IDs representing the decoded sequence