flashlight_tensor 0.4.5

gpu/cpu tensor library focused around matrix and neural network operations
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

#[cfg(test)]
mod division{
    use flashlight_tensor::prelude::*;

    #[test]
    fn iterative_tensor_division(){
        let data: Vec<f32> = vec!{1.0, 2.0, 3.0};
        let sizes: Vec<u32> = vec!{3};

        let expected: Vec<f32> = vec!{1.0, 1.0, 1.0};

        let tensor = Tensor::from_data(&data, &sizes).unwrap();

        let result_tensor = tensor.tens_div(&tensor).unwrap();
        
        assert_eq!(result_tensor.get_data(), &expected);
    }
    
    #[test]
    fn iterative_division(){
        let data: Vec<f32> = vec!{1.0, 2.0, 3.0};
        let sizes: Vec<u32> = vec!{3};

        let expected: Vec<f32> = vec!{0.5, 1.0, 1.5};

        let tensor = Tensor::from_data(&data, &sizes).unwrap();

        let result_tensor = tensor.div(2.0);
        
        assert_eq!(result_tensor.get_data(), &expected);
    }
}