Struct Embedding 

pub struct Embedding { /* private fields */ }

Embedding layer - lookup table for discrete tokens

Implementations

impl Embedding

pub fn new(num_embeddings: usize, embedding_dim: usize) -> Self

Create new embedding layer

pub fn with_padding( num_embeddings: usize, embedding_dim: usize, padding_idx: usize, ) -> Self

Create embedding with padding index (embedding at padding_idx will be zeros)

pub fn from_pretrained(weight: Tensor, freeze: bool) -> Self

Create embedding from pretrained weights

pub fn embedding_dim(&self) -> usize

Get embedding dimension

pub fn num_embeddings(&self) -> usize

Get number of embeddings

pub fn forward_indices(&self, indices: &[usize]) -> Tensor

Forward pass with integer indices

Trait Implementations

impl Module for Embedding

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

Forward pass

fn parameters(&self) -> Vec<Tensor>

Get all trainable parameters

fn train(&mut self)

Set module to training mode

fn eval(&mut self)

Set module to evaluation mode

fn is_training(&self) -> bool

Check if in training mode

fn num_parameters(&self) -> usize

Number of trainable parameters