1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
#[macro_export]
macro_rules! impl_ilayer_loss {
() => (
fn exact_num_output_blobs(&self) -> Option<usize> { Some(1) }
fn exact_num_input_blobs(&self) -> Option<usize> { Some(1) }
fn auto_output_blobs(&self) -> bool { true }
fn loss_weight(&self, output_id: usize) -> Option<f32> {
if output_id == 0 {
Some(1f32)
} else {
None
}
}
)
}
pub use self::negative_log_likelihood::{NegativeLogLikelihood, NegativeLogLikelihoodConfig};
pub mod negative_log_likelihood;