burn-nn 0.21.0

Neural network building blocks for the Burn deep learning 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

mod binary_cross_entropy;
mod cosine_embedding;
mod cross_entropy;
mod ctc;
mod huber;
mod kldiv;
mod lp_loss;
mod mse;
mod poisson;
mod reduction;
mod rnnt;
mod smooth_l1;

pub use binary_cross_entropy::*;
pub use cosine_embedding::*;
pub use cross_entropy::*;
pub use ctc::*;
pub use huber::*;
pub use kldiv::*;
pub use lp_loss::*;
pub use mse::*;
pub use poisson::*;
pub use reduction::*;
pub use rnnt::*;
pub use smooth_l1::*;