Struct EmbeddingLayer 

pub struct EmbeddingLayer { /* private fields */ }

Embedding lookup table: maps integer indices to dense vectors.

Implementations

impl EmbeddingLayer

pub fn new( graph: &mut Graph, num_embeddings: usize, embedding_dim: usize, weight_init: Tensor, ) -> Result<Self, ModelError>

pub fn num_embeddings(&self) -> usize

pub fn embedding_dim(&self) -> usize

pub fn weight_node(&self) -> NodeId

pub fn forward( &self, graph: &mut Graph, input: NodeId, ) -> Result<NodeId, ModelError>

pub fn forward_inference( &self, graph: &Graph, indices: &Tensor, ) -> Result<Tensor, ModelError>

Trait Implementations

impl Clone for EmbeddingLayer

fn clone(&self) -> EmbeddingLayer

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for EmbeddingLayer

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

Formats the value using the given formatter.