rotta_rs 0.1.0

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
38
39
40

use crate::{ rotta_rs_module::{ arrayy::{ ArrSlice, Arrayy } }, ShareTensor };

#[derive(Debug, Clone)]
pub enum BackwardLabel {
    // operation
    Dot(ShareTensor, ShareTensor),
    Matmul(ShareTensor, ShareTensor),
    Add(ShareTensor, ShareTensor),
    Diveded(ShareTensor, ShareTensor),
    Mul(ShareTensor, ShareTensor),
    Sub(ShareTensor, ShareTensor),

    // mutation
    Index(ShareTensor, Vec<i32>),
    Broadcasting(ShareTensor, Arrayy),
    SumAxis(ShareTensor, Vec<i32>, bool), // keep_dim
    Sum(ShareTensor),
    Permute(ShareTensor, Vec<usize>),
    Slice(ShareTensor, Vec<ArrSlice>),
    ToShape(ShareTensor, Vec<usize>),
    Concat(Vec<ShareTensor>, usize),

    // method
    Exp(ShareTensor, Arrayy),
    Powi(ShareTensor, i32),
    Powf(ShareTensor, f32),
    Ln(ShareTensor),
    Abs(ShareTensor),
    Sign(ShareTensor),
    Sin(ShareTensor),
    Cos(ShareTensor),
    Tan(ShareTensor),

    // activation
    Relu(ShareTensor),

    // loss
    SSResidual(ShareTensor, ShareTensor), // prediction, actual
    CEL(ShareTensor, ShareTensor), // prediction, actual
}