Enum PoolingStrategy 

pub enum PoolingStrategy {
    Mean,
    Cls,
}

Pooling strategy for combining token embeddings into a sentence embedding.

Variants

Mean

Mean pooling over non-padding tokens.

Cls

Use the [CLS] token (position 0) embedding.

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 Copy for PoolingStrategy

impl Eq for PoolingStrategy

impl StructuralPartialEq for PoolingStrategy