yarnn 0.1.0

Yet Another rust Neural Network framework
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

use crate::loss::Loss;
use crate::backend::{Backend, BackendSub};
use core::marker::PhantomData;



pub struct CrossEntropyLoss<N, B> {
    _m: PhantomData<fn(N, B)>   
}

impl<N, B> CrossEntropyLoss<N, B> {
    pub fn new() -> Self {
        Self {
            _m: Default::default()
        }
    }
}

impl<N, B: Backend<N> + BackendSub<N>> Loss<N, B> for CrossEntropyLoss<N, B> {
    fn compute(&self, _backend: &B, _dst: &mut B::Tensor, _pred: &B::Tensor, _target: &B::Tensor) {
        // TODO
    }

    fn derivative(&self, backend: &B, dst: &mut B::Tensor, pred: &B::Tensor, target: &B::Tensor) {      
        backend.sub(dst, pred, target);
    }
}