Enum HeadAggregation

pub enum HeadAggregation {
    None,
    Mean,
    Max,
    WeightedMean(Vec<f64>),
    Rollout,
}

Head aggregation methods

Variants

None

No aggregation

Mean

Average across heads

Max

Maximum across heads

WeightedMean(Vec<f64>)

Weighted average

Rollout

Attention rollout

Trait Implementations

impl Clone for HeadAggregation

fn clone(&self) -> HeadAggregation

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for HeadAggregation

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

Formats the value using the given formatter.

impl PartialEq for HeadAggregation

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

Tests for self and other values to be equal, and is used by ==.

const 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 HeadAggregation

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

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl StructuralPartialEq for HeadAggregation