newron 0.5.1

A Rust library to train and infer deep learning models.
Documentation 
1
2
3
4
5
6

use crate::tensor::Tensor;

pub trait Loss {
    fn compute_loss(&self, y_true: &Tensor, y_pred: &Tensor) -> f64;
    fn compute_loss_grad(&self, y_true: &Tensor, y_pred: &Tensor) -> Tensor;
}