embed_token

avarok_core::kimi_k3::ops

Function embed_token 

Source 
pub fn embed_token(
    table: &[f32],
    token: u32,
    hidden: usize,
    vocab: usize,
) -> Vec<f32>
Expand description

Token embedding gather: embed[token, :].