Class that specifies parameters of the batch normalization layer.
More...
Constructs the parameters for the batch normalization layer
Gets the smoothing factor of the batch normalization layer
Gets the index of the dimension for which the normalization is performed
Gets the constant added to the mini-batch variance for numerical stability
boolean getPredictionStage |
( |
| ) |
|
Gets the flag that specifies whether the layer is used for the prediction stage or not
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 |
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 |
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 |
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