Skip to main content

cumsum

Function cumsum 

Source
pub fn cumsum(
    a: &Array,
    axis: Option<isize>,
    reverse: bool,
    inclusive: bool,
) -> Result<Array>
Expand description

The cumulative sum along axis (flattened without one): reverse scans from the end; an exclusive scan (inclusive = false) leaves each element out of its own output. Booleans sum as int32.

use nuvec_core::{Array, Result, cumsum};

let a = Array::from_slice(&[1i32, 2, 3, 4], &[4])?;
assert_eq!(cumsum(&a, None, false, true)?.to_vec::<i32>()?, [1, 3, 6, 10]);
assert_eq!(cumsum(&a, None, true, false)?.to_vec::<i32>()?, [9, 7, 4, 0]);