Struct RuleBasedMask 

pub struct RuleBasedMask {
    pub batch_size: usize,
    pub pattern: RulePattern,
    pub rule_spec: String,
}

Rule-based attention mask

Fields

batch_size: usize

Batch size

pattern: RulePattern

Pattern type

rule_spec: String

Rule specification (opaque for now)

Implementations

impl RuleBasedMask

pub fn new(batch_size: usize, pattern: RulePattern, rule_spec: String) -> Self

Create a new rule-based mask

pub fn hard(batch_size: usize, rule_spec: String) -> Self

Create a hard rule-based mask

pub fn soft(batch_size: usize, rule_spec: String) -> Self

Create a soft rule-based mask

pub fn gated(batch_size: usize, rule_spec: String) -> Self

Create a gated rule-based mask

Trait Implementations

impl AttentionMask for RuleBasedMask

fn build_mask(&self, graph: &mut EinsumGraph, seq_len: usize) -> Result<usize>

Build a mask tensor in the einsum graph

fn mask_type(&self) -> &str

Get mask type name for documentation

impl Clone for RuleBasedMask

fn clone(&self) -> RuleBasedMask

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for RuleBasedMask

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

Formats the value using the given formatter.