flashlight 0.0.12

neural network library
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

use flashlight_tensor::tensor::Tensor;
use async_trait::async_trait;

pub mod linear;
pub mod dropout;

pub mod activations;

pub trait Backend{
    fn forward(&mut self, input: Tensor<f32>){}
    fn backward(&mut self, grad_output: Tensor<f32>){}
}

pub struct Cpu;
pub struct Gpu;

/// Basic layer trait for layers
pub trait LayerCpu {
    /// Forward propagation for layer
    fn forward(&mut self, input: &Tensor<f32>) -> Tensor<f32>;
    /// Backward propagation for layer
    fn backward(&mut self, grad_output: &Tensor<f32>) -> Tensor<f32>;
}

#[async_trait]
pub trait LayerGpu {
    /// Forward propagation for layer
    async fn forward(&mut self, input: &Tensor<f32>) -> Tensor<f32>;
    /// Backward propagation for layer
    async fn backward(&mut self, grad_output: &Tensor<f32>) -> Tensor<f32>;
}