Enum neuronika::nn::loss::Reduction [−][src]
pub enum Reduction {
Sum,
Mean,
}
Expand description
Specifies the reduction to apply to the loss output.
Variants
Sum
The output will be summed.
Mean
The sum of the output will be divided by the batch size for the kldiv_loss
and the
nll_loss
. For all other losses the output will be divided by the number of elements.
Trait Implementations
Auto Trait Implementations
impl RefUnwindSafe for Reduction
impl UnwindSafe for Reduction
Blanket Implementations
Mutably borrows from an owned value. Read more