ferrite-dl 0.2.0

Deep learning library written in pure Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

use super::loss::*;
use crate::tensor::*;

pub struct CrossEntropyLoss {
  
}

impl CrossEntropyLoss {
  pub fn new() -> Self {
    CrossEntropyLoss { }
  }
}

impl LossTrait for CrossEntropyLoss {
  fn loss(&self, x: &Tensor, y: &Tensor) -> Tensor {
    unimplemented!()

  }
}