broadcasting/
broadcasting.rs

1use numrst::NdArray;
2
3fn main() -> Result<(), Box<dyn std::error::Error>> {
4    let a = NdArray::<f32>::ones((3, 3))?;
5    let b = NdArray::<f32>::arange(0.0, 3.0)?.reshape((3, 1))?;
6
7    let c = a.broadcast_add(&b)?; // broadcasting
8    println!("a:\n{}", a);
9    println!("b:\n{}", b);
10    println!("a + b (broadcast):\n{}", c);
11
12    Ok(())
13}