Struct syntaxdot_transformers::layers::Embedding

pub struct Embedding(pub Tensor);

Embedding lookup layer.

Tuple Fields

0: Tensor

Implementations

impl Embedding

pub fn new<'a>(
    vs: impl Borrow<PathExt<'a>>,
    name: &str,
    num_embeddings: i64,
    embedding_dim: i64,
    init: Init
) -> Result<Self, TransformerError>

Trait Implementations

impl Debug for Embedding

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

Formats the value using the given formatter.

impl FallibleModule for Embedding

type Error = TransformerError

The error type.

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

Apply the module.