burn-nn 0.22.0

Neural network building blocks for the Burn deep learning framework
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
26
27
28
29
30
31
32
33
34
35
36
37

mod binary_cross_entropy;
mod cosine_embedding;
mod cross_entropy;
mod ctc;
mod gaussian_nll;
mod hinge_embedding;
mod huber;
mod kldiv;
mod lp_loss;
mod margin_ranking;
mod mse;
mod multi_margin;
mod poisson;
mod reduction;
mod rnnt;
mod smooth_l1;
mod soft_margin;
mod triplet_margin;

pub use binary_cross_entropy::*;
pub use cosine_embedding::*;
pub use cross_entropy::*;
pub use ctc::*;
pub use gaussian_nll::*;
pub use hinge_embedding::*;
pub use huber::*;
pub use kldiv::*;
pub use lp_loss::*;
pub use margin_ranking::*;
pub use mse::*;
pub use multi_margin::*;
pub use poisson::*;
pub use reduction::*;
pub use rnnt::*;
pub use smooth_l1::*;
pub use soft_margin::*;
pub use triplet_margin::*;