org.deeplearning4j.nn.layers.normalization

Interface BatchNormalizationHelper

public interface BatchNormalizationHelper

Helper for the batch normalization layer.

Author:

saudet

Method Summary

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

Method Detail

checkSupported

boolean checkSupported(double eps)

backpropGradient

org.nd4j.linalg.primitives.Pair<Gradient,org.nd4j.linalg.api.ndarray.INDArray> backpropGradient(org.nd4j.linalg.api.ndarray.INDArray input,
                                                                                                org.nd4j.linalg.api.ndarray.INDArray epsilon,
                                                                                                int[] shape,
                                                                                                org.nd4j.linalg.api.ndarray.INDArray gamma,
                                                                                                org.nd4j.linalg.api.ndarray.INDArray dGammaView,
                                                                                                org.nd4j.linalg.api.ndarray.INDArray dBetaView,
                                                                                                double eps,
                                                                                                LayerWorkspaceMgr workspaceMgr)

preOutput

org.nd4j.linalg.api.ndarray.INDArray preOutput(org.nd4j.linalg.api.ndarray.INDArray x,
                                               boolean training,
                                               int[] shape,
                                               org.nd4j.linalg.api.ndarray.INDArray gamma,
                                               org.nd4j.linalg.api.ndarray.INDArray beta,
                                               org.nd4j.linalg.api.ndarray.INDArray mean,
                                               org.nd4j.linalg.api.ndarray.INDArray var,
                                               double decay,
                                               double eps,
                                               LayerWorkspaceMgr workspaceMgr)