pub fn compute_output_shape(
output_subs: &[u32],
size_dict: &HashMap<u32, usize>,
) -> Result<Vec<usize>>Expand description
Compute output shape from output subscripts and size dictionary.
ยงExamples
use tenferro_einsum::{build_size_dict, compute_output_shape, Subscripts};
let subs = Subscripts::parse("ij,jk->ik").unwrap();
let shapes = [&[2, 3][..], &[3, 4][..]];
let sizes = build_size_dict(&subs, &shapes, None).unwrap();
let output_shape = compute_output_shape(&subs.output, &sizes).unwrap();
assert_eq!(output_shape, vec![2, 4]);