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

use crate::{ rotta_rs_module::{ arrayy::Arrayy, BackwardLabel, Tensor }, ShareTensor };

pub fn powf(x: &Tensor, n: f32) -> Tensor {
    let arrayy = x.value.read().unwrap().powf(n);
    let mut tensor = Tensor::from_arrayy(arrayy);
    tensor.update_parent(vec![x.shared_tensor()]);
    tensor.update_label(Some(BackwardLabel::Powf(x.shared_tensor(), n)));

    tensor
}

pub fn d_powf(x: &ShareTensor, powf: f32, grad: &Arrayy) {
    // d/x = n * x^n-1
    if x.requires_grad() {
        let dx =
            powf *
            &x.value
                .read()
                .unwrap()
                .powf(powf - 1.0) *
            grad;

        x.add_grad(dx);
    }
}