Enum PoolingStrategy 

pub enum PoolingStrategy {
    Mean,
    Max,
    CLS,
    AttentionWeighted,
}

Pooling strategy for combining token embeddings.

Variants

Mean

Average of all token embeddings.

Max

Max pooling across tokens.

CLS

Use CLS token embedding (first token).

AttentionWeighted

Weighted average by attention.

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<'de> Deserialize<'de> for PoolingStrategy

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

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

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl Copy for PoolingStrategy

impl Eq for PoolingStrategy

impl StructuralPartialEq for PoolingStrategy