Function burn::nn::attention::generate_padding_mask

pub fn generate_padding_mask<B>(
    pad_token: usize,
    tokens_list: Vec<Vec<usize>>,
    max_seq_length: Option<usize>,
    device: &<B as Backend>::Device,
) -> GeneratePaddingMask<B>
where
    B: Backend,
Expand description

Generation padding attention mask.