org.deeplearning4j.nn.layers.normalization

Interface BatchNormalizationHelper

All Superinterfaces:

LayerHelper

public interface BatchNormalizationHelper
extends LayerHelper

Helper for the batch normalization layer.

Author:

saudet

Method Summary

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

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

helperMemoryUse

Method Detail

checkSupported

boolean checkSupported(double eps)

backpropGradient

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

preOutput

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

getMeanCache

INDArray getMeanCache()

getVarCache

INDArray getVarCache()