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

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

pub fn permute(x: &Tensor, order: Vec<usize>) -> Tensor {
    let mut tensor = Tensor::from_arrayy(x.value.read().unwrap().permute(&order));
    tensor.update_parent(vec![x.shared_tensor()]);
    tensor.update_label(Some(BackwardLabel::Permute(x.shared_tensor(), order)));

    tensor
}

pub fn d_permute(x: &ShareTensor, order: Vec<usize>, grad: &Arrayy) {
    // let x = x.read().unwrap();

    let mut new_order = order.clone();
    for (i, d) in order.iter().enumerate() {
        new_order[*d] = i;
    }

    let d_x = grad.permute(&new_order);
    x.add_grad(d_x);
}