Struct PoolingConfig

pub struct PoolingConfig {
    pub word_embedding_dimension: i64,
    pub pooling_mode_cls_token: bool,
    pub pooling_mode_max_tokens: bool,
    pub pooling_mode_mean_tokens: bool,
    pub pooling_mode_mean_sqrt_len_tokens: bool,
}

Configuration for Pooling layer.

Fields

word_embedding_dimension: i64

Dimensions for the word embeddings

pooling_mode_cls_token: bool

Use the first token (CLS token) as text representations

pooling_mode_max_tokens: bool

Use max in each dimension over all tokens

pooling_mode_mean_tokens: bool

Perform mean-pooling

pooling_mode_mean_sqrt_len_tokens: bool

Perform mean-pooling, but devide by sqrt(input_length)

Trait Implementations

impl Config for PoolingConfig

fn from_file<P: AsRef<Path>>(path: P) -> Self

Loads a Config object from a JSON file. The format is expected to be aligned with the Transformers library configuration files for each model. The parsing will fail if non-optional keys expected by the model are missing.

impl Debug for PoolingConfig

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

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for PoolingConfig

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

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.