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
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37

use crate::backend::{Backend, DefaultBackend};
use crate::layer::Layer;
use ndarray::Dimension;
use serde::{Deserialize, Serialize};
use std::marker::PhantomData;

#[derive(Serialize, Deserialize)]
pub struct EmptyLayer<D, B: Backend = DefaultBackend> {
    _phantom: PhantomData<(D, B)>,
}

impl<D, B: Backend> Default for EmptyLayer<D, B> {
    fn default() -> Self {
        Self {
            _phantom: PhantomData,
        }
    }
}

impl<D, B: Backend> EmptyLayer<D, B> {
    pub fn new() -> Self {
        Self::default()
    }
}

impl<D: Dimension, B: Backend> Layer<B> for EmptyLayer<D, B> {
    type Input = D;
    type Output = D;

    fn forward(&mut self, input: &B::Tensor<D>) -> B::Tensor<D> {
        input.clone()
    }

    fn backward(&mut self, grad_output: &B::Tensor<D>) -> B::Tensor<D> {
        grad_output.clone()
    }
}