org.deeplearning4j.nn.layers.normalization

Interface BatchNormalizationHelper

All Superinterfaces:

LayerHelper

All Known Implementing Classes:

MKLDNNBatchNormHelper

public interface BatchNormalizationHelper
extends LayerHelper

Method Summary

Modifier and Type	Method and Description
Pair<Gradient,INDArray>	backpropGradient(INDArray input, INDArray epsilon, long[] shape, INDArray gamma, INDArray beta, INDArray dGammaView, INDArray dBetaView, double eps, CNN2DFormat format, LayerWorkspaceMgr workspaceMgr)
boolean	checkSupported(double eps, boolean fixedGammaBeta)
INDArray	getMeanCache(DataType dataType)
INDArray	getVarCache(DataType dataType)
INDArray	preOutput(INDArray x, boolean training, long[] shape, INDArray gamma, INDArray beta, INDArray mean, INDArray var, double decay, double eps, CNN2DFormat format, LayerWorkspaceMgr workspaceMgr)

Methods inherited from interface org.deeplearning4j.nn.layers.LayerHelper

checkSupported, helperMemoryUse

Method Detail

checkSupported

boolean checkSupported(double eps,
                       boolean fixedGammaBeta)

backpropGradient

Pair<Gradient,INDArray> backpropGradient(INDArray input,
                                         INDArray epsilon,
                                         long[] shape,
                                         INDArray gamma,
                                         INDArray beta,
                                         INDArray dGammaView,
                                         INDArray dBetaView,
                                         double eps,
                                         CNN2DFormat format,
                                         LayerWorkspaceMgr workspaceMgr)

preOutput

INDArray preOutput(INDArray x,
                   boolean training,
                   long[] shape,
                   INDArray gamma,
                   INDArray beta,
                   INDArray mean,
                   INDArray var,
                   double decay,
                   double eps,
                   CNN2DFormat format,
                   LayerWorkspaceMgr workspaceMgr)

getMeanCache

INDArray getMeanCache(DataType dataType)

getVarCache

INDArray getVarCache(DataType dataType)