org.deeplearning4j.nn.layers.normalization

Interface BatchNormalizationHelper

All Known Implementing Classes:

CudnnBatchNormalizationHelper

public interface BatchNormalizationHelper

Helper for the batch normalization layer.

Method Summary

Modifier and Type	Method and Description
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)
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)

Method Detail

backpropGradient

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)

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)