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

use burn_core as burn;

use burn::config::Config;

/// The reduction type for the loss.
#[derive(Config, Debug)]
pub enum Reduction {
    /// The mean of the losses will be returned.
    Mean,

    /// The sum of the losses will be returned.
    Sum,

    /// The sum of the losses divided by the batch_size will be returned.
    BatchMean,

    /// The mean of the losses will be returned.
    Auto,
}