pub fn binary_ops_shape(lhs: &[usize], rhs: &[usize]) -> Vec<usize>
Computes the output shape for binary operations with broadcasting support.