jiro_nn 0.8.1

Neural Networks framework with model building & data preprocessing features.
Documentation 
1
2
3
4
5
6
7
8
9

use super::ActivationLayer;
use crate::linalg::{Matrix, MatrixTrait};

pub fn new() -> ActivationLayer {
    ActivationLayer::new(
        |m| m.maxof(&Matrix::constant(m.dim().0, m.dim().1, 0.)),
        |m| m.sign().maxof(&Matrix::constant(m.dim().0, m.dim().1, 0.)),
    )
}