Function broadcast_shapes 

pub fn broadcast_shapes(
    op: &'static str,
    a: &[usize],
    b: &[usize],
) -> Result<Vec<usize>>

The broadcast shape of a and b (NumPy rules), or op’s shape-mismatch error, as the binary operations compute it.