Skip to main content

create_padding_mask

Function create_padding_mask 

Source
pub fn create_padding_mask(
    batch_size: usize,
    seq_len: usize,
    actual_lengths: &[usize],
) -> Array3<bool>
Expand description

Helper function to create padding mask