jiro_nn 0.5.0

Neural Networks framework with model specification & data preprocessing features.
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 super::ConvActivationLayer;
use crate::vision::{image::Image, image::ImageTrait};

fn tanh(m: &Image) -> Image {
    let exp = m.exp();
    let exp_neg = m.scalar_mul(-1.).exp();
    (exp.component_sub(&exp_neg)).component_div(&(exp.component_add(&exp_neg)))
}

fn tanh_prime(m: &Image) -> Image {
    let hbt = tanh(m);
    let hbt2 = &hbt.square();
    let ones = Image::constant(
        hbt.image_dims().0,
        hbt.image_dims().1,
        hbt.channels(),
        hbt.samples(),
        1.,
    );
    ones.component_sub(&hbt2)
}

pub fn new() -> ConvActivationLayer {
    ConvActivationLayer::new(tanh, tanh_prime)
}