Struct MaskBuilder 

pub struct MaskBuilder { /* private fields */ }

Builder for attention masks

Implementations

impl MaskBuilder

pub fn new(num_heads: usize, num_layers: usize) -> Self

Create a new builder

pub fn sparsity(self, sparsity: f32) -> Self

Set target sparsity

pub fn pattern(self, pattern: MaskPattern) -> Self

Set mask pattern

pub fn min_active(self, min: usize) -> Self

Set minimum active heads

pub fn category_weights(self, weights: HashMap<HeadCategory, f32>) -> Self

Set category weights for category-based masking

pub fn category_mapping(self, mapping: CategoryMapping) -> Self

Set category mapping

pub fn build(self) -> AttentionMask

Build the mask

Trait Implementations

impl Clone for MaskBuilder

fn clone(&self) -> MaskBuilder

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for MaskBuilder

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

Formats the value using the given formatter.