Struct Embedding 

pub struct Embedding { /* private fields */ }

Implementations

impl Embedding

pub fn new( num_embeddings: u32, embedding_dim: u32, padding_idx: Option<u32>, max_norm: Option<impl Into<Scalar>>, norm_type: impl Into<Scalar>, dtype: DType, ) -> Result<Embedding, HoduError>

pub fn from_pretrained( weight: Tensor, freeze: bool, ) -> Result<Embedding, HoduError>

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

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

pub fn weight(&self) -> &Tensor

pub fn num_embeddings(&self) -> u32

pub fn embedding_dim(&self) -> u32

pub fn padding_idx(&self) -> Option<u32>

pub fn max_norm(&self) -> Option<f32>

pub fn norm_type(&self) -> f32

Trait Implementations

impl Clone for Embedding

fn clone(&self) -> Embedding

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Module<&Tensor> for Embedding

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

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