Struct EmbeddingBagMaxDescriptor 

pub struct EmbeddingBagMaxDescriptor {
    pub num_embeddings: i32,
    pub embedding_dim: i32,
    pub num_bags: i32,
    pub total_indices: i32,
    pub padding_idx: Option<i32>,
    pub element: ElementKind,
}

Descriptor for an embedding_bag Max-mode op.

Fields

num_embeddings: i32

Vocabulary size — extent of weight along axis 0.

embedding_dim: i32

Embedding dimension — extent of weight along axis 1.

num_bags: i32

Number of bags — extent of offsets and of out, out_index along axis 0.

total_indices: i32

Total flat-index length — extent of indices.

padding_idx: Option<i32>

Optional padding index. Indices matching p (or negative / OOB) are dropped from the bag.

element: ElementKind

Value element type.

Trait Implementations

impl Clone for EmbeddingBagMaxDescriptor

fn clone(&self) -> EmbeddingBagMaxDescriptor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for EmbeddingBagMaxDescriptor

impl Debug for EmbeddingBagMaxDescriptor

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

Formats the value using the given formatter.