Skip to main content

broadcast_shape

Function broadcast_shape 

Source
pub fn broadcast_shape(
    lhs: &[usize],
    rhs: &[usize],
) -> Result<ShapeVec, 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().as_slice(), &[3, 4]);
assert_eq!(broadcast_shape(&[], &[2, 3]).unwrap().as_slice(), &[2, 3]);

§Errors

Returns BroadcastError::IncompatibleBinary when a pair of aligned dimensions is incompatible.