Function broadcast_strides 

pub fn broadcast_strides(
    n_new_batch: usize,
    rank_prev: usize,
    num_elems: usize,
    strides: &[usize],
) -> Vec<usize>

Calculate the new strides given added batch dimensions.