Enum PoolingStrategy 

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

Pooling strategies for transformer outputs

Variants

Cls

Use [CLS] token embedding

Mean

Mean pooling of all token embeddings

Max

Max pooling of all token embeddings

AttentionWeighted

Weighted mean pooling based on attention weights

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 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.