ferrite-dl 0.2.0

Deep learning library written in pure Rust
Documentation
1
2
3
4
5
6
use crate::tensor::*;

pub trait LossTrait {
  fn loss(&self, x: &Tensor, y: &Tensor) -> Tensor;
}