Struct CharacterEmbedding 

pub struct CharacterEmbedding {
    pub weight: Array2<f64>,
    /* private fields */
}

Trainable character embedding layer.

Maps character indices to dense vectors.

Fields

weight: Array2<f64>

Implementations

impl CharacterEmbedding

pub fn new(vocab_size: usize, embed_dim: usize) -> Self

Create new embedding with random initialization.

pub fn new_zeros(vocab_size: usize, embed_dim: usize) -> Self

Create embedding with zero initialization.

pub fn from_weights(weight: Array2<f64>) -> Self

Create embedding from existing weights.

pub fn embed_dim(&self) -> usize

Get embedding dimension.

pub fn vocab_size(&self) -> usize

Get vocabulary size.

pub fn lookup(&self, char_idx: usize) -> Array1<f64>

Lookup single character embedding.

pub fn forward(&mut self, char_indices: &[usize]) -> Array2<f64>

Forward pass for sequence of indices. Returns (seq_len, embed_dim) matrix.

pub fn backward(&self, grad_output: &Array2<f64>) -> EmbeddingGradients

Backward pass - compute gradients. grad_output shape: (seq_len, embed_dim)

pub fn update_parameters<O: Optimizer>( &mut self, gradients: &EmbeddingGradients, optimizer: &mut O, prefix: &str, )

Update parameters with optimizer.

pub fn num_parameters(&self) -> usize

Get number of parameters.

Trait Implementations

impl Clone for CharacterEmbedding

fn clone(&self) -> CharacterEmbedding

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for CharacterEmbedding

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.