Skip to main content

compute_output_shape

Function compute_output_shape 

Source
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]);