Java* API Reference for Intel® Data Analytics Acceleration Library 2019 Update 5

List of all members
BatchNormalizationParameter Class Reference

Class that specifies parameters of the batch normalization layer. More...

Class Constructor

Constructs the parameters for the batch normalization layer

Detailed Description

Member Function Documentation

double getAlpha ( )

Gets the smoothing factor of the batch normalization layer

long getDimension ( )

Gets the index of the dimension for which the normalization is performed

double getEpsilon ( )

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
alphaSmoothing 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
dimensionBatchNormalizationIndex 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
epsilonA 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
predictionStageFlag that specifies whether the layer is used for the prediction stage or not

The documentation for this class was generated from the following file:

For more complete information about compiler optimizations, see our Optimization Notice.