yarnn 0.1.0

Yet Another rust Neural Network framework
Documentation 
1
2
3
4
5
6

use crate::backend::Backend;

pub trait Loss<N, B: Backend<N>> {
    fn compute(&self, backend: &B, dst: &mut B::Tensor, pred: &B::Tensor, target: &B::Tensor);
    fn derivative(&self, backend: &B, dst: &mut B::Tensor, pred: &B::Tensor, target: &B::Tensor);
}