Skip to main content

generate_padding_mask

yscv_model

Function generate_padding_mask 

Source 
pub fn generate_padding_mask(
    lengths: &[usize],
    max_len: usize,
) -> Result<Tensor, ModelError>
Expand description

Generates a padding mask for batched sequences with different lengths. lengths: actual length of each sequence in the batch max_len: maximum sequence length (pad length) Returns [batch, max_len] tensor where:

  • 0.0 for valid positions (index < length)
  • f32::NEG_INFINITY for padding positions (index >= length)