Package org.jetbrains.kotlinx.dl.api.core.layer.normalization

Types

BatchNorm

class BatchNorm(axis: List<Int>, momentum: Double, center: Boolean, epsilon: Double, scale: Boolean, gammaInitializer: Initializer, betaInitializer: Initializer, gammaRegularizer: Regularizer?, betaRegularizer: Regularizer?, movingMeanInitializer: Initializer, movingVarianceInitializer: Initializer, name: String) : Layer, NoGradients, ParametrizedLayer

NOTE: This layer is not trainable and does not update its weights. It's frozen by default.