meuron 0.4.0

Meuron is a modular neural network library written in rust for training simple neural networks.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21

pub mod relu;
pub mod sigmoid;
pub mod softmax;
pub mod tanh;

pub use relu::ReLU;
pub use sigmoid::Sigmoid;
pub use softmax::Softmax;
pub use tanh::Tanh;

use crate::backend::Backend;
use ndarray::Dimension;

pub trait Activation<B: Backend>: Clone {
    fn activate<D: Dimension>(&self, x: &B::Tensor<D>) -> B::Tensor<D>;
    fn derivative<D: Dimension>(&self, x: &B::Tensor<D>) -> B::Tensor<D>;

    fn vjp<D: Dimension>(&self, z: &B::Tensor<D>, grad: &B::Tensor<D>) -> B::Tensor<D> {
        B::mul(grad, &self.derivative(z))
    }
}