Class that specifies parameters of the batch normalization layer.
More...
◆ BatchNormalizationParameter()
Constructs the parameters for the batch normalization layer
◆ getAlpha()
Gets the smoothing factor of the batch normalization layer
◆ getDimension()
Gets the index of the dimension for which the normalization is performed
◆ getEpsilon()
Gets the constant added to the mini-batch variance for numerical stability
◆ getPredictionStage()
boolean getPredictionStage |
( |
| ) |
|
Gets the flag that specifies whether the layer is used for the prediction stage or not
◆ setAlpha()
void setAlpha |
( |
double |
alpha | ) |
|
Sets the smoothing factor of the batch normalization layer
- Parameters
-
alpha | Smoothing factor that is used in population mean and population variance computations |
◆ setDimension()
void setDimension |
( |
long |
dimension | ) |
|
Sets the index of the dimension for which the normalization is performed
- Parameters
-
dimension | BatchNormalizationIndex of the dimension for which the normalization is performed |
◆ setEpsilon()
void setEpsilon |
( |
double |
epsilon | ) |
|
Sets the constant added to the mini-batch variance for numerical stability
- Parameters
-
epsilon | A constant added to the mini-batch variance for numerical stability |
◆ setPredictionStage()
void setPredictionStage |
( |
boolean |
predictionStage | ) |
|
Sets the flag that specifies whether the layer is used for the prediction stage or not
- Parameters
-
predictionStage | Flag that specifies whether the layer is used for the prediction stage or not |
The documentation for this class was generated from the following file:
- BatchNormalizationParameter.java