Struct BlockSparseMask 

pub struct BlockSparseMask {
    pub batch_size: usize,
    pub block_size: usize,
}

Block-sparse attention mask

Divides attention into fixed-size blocks (used in BigBird, Longformer).

Fields

batch_size: usize

Batch size

block_size: usize

Block size

Implementations

impl BlockSparseMask

pub fn new(batch_size: usize, block_size: usize) -> Result<Self>

Create a new block-sparse attention mask

Trait Implementations

impl AttentionMask for BlockSparseMask

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 BlockSparseMask

fn clone(&self) -> BlockSparseMask

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for BlockSparseMask

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

Formats the value using the given formatter.