org.deeplearning4j.nn.layers.normalization

Class CudnnBatchNormalizationHelper

java.lang.Object

org.deeplearning4j.nn.layers.normalization.CudnnBatchNormalizationHelper

All Implemented Interfaces:

BatchNormalizationHelper

public class CudnnBatchNormalizationHelper
extends java.lang.Object
implements BatchNormalizationHelper

cuDNN-based helper for the batch normalization layer.

Field Summary

Fields

Modifier and Type	Field and Description
protected static org.slf4j.Logger	log

Constructor Summary

Constructors

Constructor and Description
CudnnBatchNormalizationHelper()

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)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Detail

log

protected static final org.slf4j.Logger log

Constructor Detail

CudnnBatchNormalizationHelper

public CudnnBatchNormalizationHelper()

Method Detail

backpropGradient

public 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)

Specified by:

backpropGradient in interface BatchNormalizationHelper

preOutput

public 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)

Specified by:

preOutput in interface BatchNormalizationHelper