Struct Embedding 

pub struct Embedding { /* private fields */ }

Embedding layer: maps integer indices to dense vectors.

Weight has shape [vocab_size, embed_dim]. Input indices are 1D [seq_len] (stored as f32, cast to usize internally). Output is [seq_len, embed_dim].

Implementations

impl Embedding

pub fn new(weight: Tensor) -> Self

Create a new Embedding from a pre-existing weight tensor [vocab_size, embed_dim].

pub fn weight(&self) -> &Tensor

Get a reference to the weight tensor.

Trait Implementations

impl Module for Embedding

fn forward(&self, input: &Tensor) -> Result<Tensor>

Run the forward pass.