Enum PoolingStrategy 

#[non_exhaustive]
pub enum PoolingStrategy {
    MeanPooling,
    ClsPooling,
    MaxPooling,
    WeightedMeanPooling,
}

Strategy for aggregating per-token embeddings into a sentence vector.

Variants (Non-exhaustive)

Non-exhaustive enums could have additional variants added in future. Therefore, when matching against variants of non-exhaustive enums, an extra wildcard arm must be added to account for any future variants.

MeanPooling

Average of all token embeddings (padding token 0 is excluded).

ClsPooling

Use only the embedding of the first token (CLS-style).

MaxPooling

Element-wise maximum across all token embeddings.

WeightedMeanPooling

Weighted mean: earlier tokens receive linearly higher weight (a triangular weighting scheme).

Trait Implementations

impl Clone for PoolingStrategy

fn clone(&self) -> PoolingStrategy

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for PoolingStrategy

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

Formats the value using the given formatter.

impl PartialEq for PoolingStrategy

fn eq(&self, other: &PoolingStrategy) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Eq for PoolingStrategy

impl StructuralPartialEq for PoolingStrategy