jiro_nn 0.8.0

Neural Networks framework with model building & 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};

pub fn new() -> ConvActivationLayer {
    ConvActivationLayer::new(
        |m| {
            m.maxof(&Image::constant(
                m.image_dims().0,
                m.image_dims().1,
                m.channels(),
                m.samples(),
                0.,
            ))
        },
        |m| {
            m.sign().maxof(&Image::constant(
                m.image_dims().0,
                m.image_dims().1,
                m.channels(),
                m.samples(),
                0.,
            ))
        },
    )
}