Function broadcast_shapes 

pub fn broadcast_shapes(
    shape1: &[usize],
    shape2: &[usize],
) -> Option<Vec<usize>>

Calculate output shape after broadcasting