Struct candle_nn::embedding::Embedding

pub struct Embedding { /* private fields */ }

Implementations

impl Embedding

pub fn new(embeddings: Tensor, hidden_size: usize) -> Self

pub fn embeddings(&self) -> &Tensor

pub fn hidden_size(&self) -> usize

Get the hidden size of the embedding matrix

Trait Implementations

impl Clone for Embedding

fn clone(&self) -> Embedding

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Debug for Embedding

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

Formats the value using the given formatter.

impl Module for Embedding

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