Enum Pooling 

pub enum Pooling {
    Mean,
    Cls,
}

How the model turns a sequence of token vectors into one sentence vector.

Part of the embedding-space identity rather than a tuning knob: the same weights pooled two ways produce vectors of the same width that must never be compared.

Variants

Mean

Average over the tokens the attention mask keeps. What the sentence-transformers MiniLM models are trained with.

Cls

The first token’s vector.

Trait Implementations

impl Clone for Pooling

fn clone(&self) -> Self

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Copy for Pooling

impl Debug for Pooling

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

Formats the value using the given formatter.

impl Default for Pooling

fn default() -> Self

Returns the “default value” for a type.

impl Eq for Pooling

impl PartialEq for Pooling

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for Pooling