matten 0.20.6

A family-car multidimensional array (tensor) library for small numerical trials / PoCs.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33

//! Rolling window operations: sum and max over overlapping slices.
//!
//! Run: cargo run --example rolling_windows_basic

use matten::Tensor;

fn rolling_sum(t: &Tensor, window: usize) -> Vec<f64> {
    (0..=(t.len() - window))
        .map(|i| t.slice().range(i..i + window).build().unwrap().sum())
        .collect()
}

fn rolling_max(t: &Tensor, window: usize) -> Vec<f64> {
    (0..=(t.len() - window))
        .map(|i| t.slice().range(i..i + window).build().unwrap().max())
        .collect()
}

fn main() {
    let series = Tensor::from_vec(vec![3.0, 1.0, 4.0, 1.0, 5.0, 9.0, 2.0, 6.0]);
    let w = 3;

    let sums = rolling_sum(&series, w);
    let maxes = rolling_max(&series, w);

    println!("series       = {:?}", series.as_slice());
    println!("rolling sum  = {:?}", sums);
    println!("rolling max  = {:?}", maxes);

    assert_eq!(sums[0], 8.0); // 3+1+4
    assert_eq!(maxes[0], 4.0); // max(3,1,4)
    println!("Rolling windows: OK");
}