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