Struct Embedding 

pub struct Embedding {
    pub weight: Tensor,
}

Token embedding table.

Fields

weight: Tensor

Embedding table [vocab, hidden].

Implementations

impl Embedding

pub fn new(weight: Tensor) -> Self

Wraps an embedding table.

pub fn forward( &self, ctx: &WgpuContext, _engine: &ComputeEngine, input_ids: &[u32], ) -> Tensor

Looks up input_ids and returns [1, seq, hidden]. The lookup runs on the CPU (the table is read back from the GPU).

pub fn params(&self) -> Vec<Tensor>

Trainable tensors (weights, then any LoRA adapters).

pub fn named_params(&self, prefix: &str) -> HashMap<String, Tensor>

Parameters keyed by Hugging Face-style names under prefix.