pub fn create_padding_mask( batch_size: usize, seq_len: usize, actual_lengths: &[usize], ) -> Array3<bool>
Helper function to create padding mask