Java* API Reference for Intel® Data Analytics Acceleration Library 2018 Update 2

List of all members
BatchNormalizationParameter Class Reference

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

Class Constructor

◆ BatchNormalizationParameter()

Constructs the parameters for the batch normalization layer

Detailed Description

Member Function Documentation

◆ getAlpha()

double getAlpha ( )

Gets the smoothing factor of the batch normalization layer

◆ getDimension()

long getDimension ( )

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

◆ getEpsilon()

double 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
alphaSmoothing 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
dimensionBatchNormalizationIndex 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
epsilonA 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
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.