sum_

sum_(
    x: &Tensor<T>, 
    dims: 
        &[i64]
        | &[i64; _]
        | [i64; _] 
        | Vec<i64> 
        | &Vec<i64>
        | i64,
    keepdim: bool,
    init_out: bool,
    out: &Tensor<T> | Tensor<T>
) -> Result<Tensor<T>, TensorError>

use hpt::{ops::NormalReduce, Tensor, error::TensorError};

fn main() -> Result<(), TensorError> {
    // Sum over dimension 0
    let a = Tensor::<f32>::new([10.0]);
    let b = Tensor::<f32>::new([0.0]);
    let c = a.sum_([0], false, false, &mut b.clone())?;
    println!("{}", c); // [10.]

    // Sum over multiple dimensions with keepdim=true
    let d = Tensor::<f32>::new([[1.0, 2.0], [3.0, 4.0]]);
    let e = Tensor::<f32>::new([[0.0]]);
    let f = d.sum_([0, 1], true, false, &mut e.clone())?;
    println!("{}", f); // [[10.]]
    Ok(())
}