Skip to main content

broadcast_shape

Function broadcast_shape 

Source
pub fn broadcast_shape(
    lhs: &[usize],
    rhs: &[usize],
) -> Result<Vec<usize>, BroadcastError>
Expand description

Compute the NumPy-style broadcast shape for two concrete shapes.

ยงExamples

use tenferro_ops::broadcast::broadcast_shape;

assert_eq!(broadcast_shape(&[3, 1], &[1, 4]).unwrap(), vec![3, 4]);
assert_eq!(broadcast_shape(&[], &[2, 3]).unwrap(), vec![2, 3]);