Struct tch::nn::Embedding

pub struct Embedding {
    pub ws: Tensor,
    /* private fields */
}

An embedding layer.

An embedding layer acts as a simple lookup table that stores embeddings. This is commonly used to store word embeddings.

Fields

ws: Tensor

Trait Implementations

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, xs: &Tensor) -> Tensor