Struct SequenceMask 

pub struct SequenceMask { /* private fields */ }

Sequence mask for variable-length batches

Tracks which positions in a batch are valid vs. padding

Implementations

impl SequenceMask

pub fn from_lengths(lengths: &[usize]) -> CoreResult<Self>

Create a new sequence mask from lengths

pub fn mask(&self) -> &Array2<bool>

Get the boolean mask

pub fn lengths(&self) -> &Array1<usize>

Get sequence lengths

pub fn max_len(&self) -> usize

Get maximum length

pub fn batch_size(&self) -> usize

Get batch size

pub fn is_valid(&self, batch_idx: usize, seq_idx: usize) -> bool

Check if a position is valid (not padding)

pub fn count_valid(&self) -> usize

Count total number of valid (non-padding) positions

Trait Implementations

impl Clone for SequenceMask

fn clone(&self) -> SequenceMask

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for SequenceMask

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

Formats the value using the given formatter.