Skip to main content

broadcast_shapes

Function broadcast_shapes 

Source
pub fn broadcast_shapes<'a>(
    shapes: impl IntoIterator<Item = &'a [usize]>,
) -> Result<Vec<usize>, BroadcastError>
Expand description

Compute the common NumPy-style broadcast shape for zero or more shapes.

ยงExamples

use tenferro_ops::broadcast::broadcast_shapes;

let shape = broadcast_shapes([&[3, 1][..], &[1, 4][..], &[3, 4][..]]).unwrap();
assert_eq!(shape, vec![3, 4]);