Struct EmbeddingBackwardDescriptor 

pub struct EmbeddingBackwardDescriptor {
    pub num_embeddings: i32,
    pub embedding_dim: i32,
    pub num_indices: i32,
    pub padding_idx: Option<i32>,
    pub element: ElementKind,
}

Descriptor for an embedding_backward op.

Fields

num_embeddings: i32

Vocabulary size — extent of dweight along axis 0.

embedding_dim: i32

Embedding dimension — extent of dweight along axis 1.

num_indices: i32

Number of indices — extent of indices and of dout along axis 0.

padding_idx: Option<i32>

Optional padding index. When Some(p), rows where indices[n] == p are skipped (no contribution).

element: ElementKind

Value element type.

Trait Implementations

impl Clone for EmbeddingBackwardDescriptor

fn clone(&self) -> EmbeddingBackwardDescriptor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for EmbeddingBackwardDescriptor

impl Debug for EmbeddingBackwardDescriptor

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

Formats the value using the given formatter.