Function find_output_shape 

pub fn find_output_shape(
    inputs: &[&str],
    shapes: &[TensorShapeType],
    output: &str,
) -> TensorShapeType

Find the output shape for given inputs, shapes and output string, taking into account broadcasting.

Examples

assert_eq!(find_output_shape(&["ab", "bc"], &[vec![2, 3], vec![3, 4]], "ac"), vec![2, 4]);
// Broadcasting is accounted for
assert_eq!(find_output_shape(&["a", "a"], &[vec![4], vec![1]], "a"), vec![4]);