rotta_rs 0.0.5

a Deep Learning library with rust language
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
34
35
36
37

use crate::arrayy::{ Arrayy, MultipleSum };

pub fn concat_arr(arrayys: Vec<Arrayy>, dim: usize) -> Arrayy {
    let mut shape = arrayys.get(0).unwrap().shape.clone();
    if dim >= shape.len() {
        panic!("can't concat cause dim:{dim} out of arrayy dimension:{}", shape.len());
    }

    let arrays_len = arrayys.len();
    let length = (&shape[dim..]).multiple_sum();

    shape[dim] = shape[dim] * arrays_len;

    let shape_len = shape.multiple_sum();
    let looping_count = shape_len / length;

    let mut arrayys_idx = 0;
    let mut t = 0;
    let mut output = Vec::with_capacity(shape_len);
    for _ in 0..looping_count {
        let vector = &arrayys[arrayys_idx].value[..];
        let start = t * length;
        let stop = start + length;

        let slice = &vector[start..stop];
        output.extend_from_slice(slice);

        arrayys_idx += 1;
        if arrayys_idx >= arrays_len {
            t += 1;
            arrayys_idx = 0;
        }
    }

    let arr = Arrayy::from_vector(shape, output);
    arr
}