rotta_rs 0.0.5

a Deep Learning library with rust language
Documentation 
1
2
3
4
5
6
7
8
9
10

use crate::rotta_rs_module::{ divided, exp, sum_axis_keep_dim, Tensor };

#[allow(dead_code)]
pub fn softmax(x: &Tensor, d: i32) -> Tensor {
    let exp = exp(x);
    let sum = sum_axis_keep_dim(&exp, &[d]);

    let s = divided(&exp, &sum);
    s
}