org.deeplearning4j.nn.layers.normalization

Class CudnnLocalResponseNormalizationHelper

java.lang.Object

org.deeplearning4j.nn.layers.normalization.CudnnLocalResponseNormalizationHelper

All Implemented Interfaces:

LocalResponseNormalizationHelper

public class CudnnLocalResponseNormalizationHelper
extends java.lang.Object
implements LocalResponseNormalizationHelper

cuDNN-based helper for the local response normalization layer.

Constructor Summary

Constructors

Constructor and Description
CudnnLocalResponseNormalizationHelper()

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.api.ndarray.INDArray	activate(org.nd4j.linalg.api.ndarray.INDArray input, boolean training, double k, double n, double alpha, double beta)
Pair<Gradient,org.nd4j.linalg.api.ndarray.INDArray>	backpropGradient(org.nd4j.linalg.api.ndarray.INDArray input, org.nd4j.linalg.api.ndarray.INDArray epsilon, double k, double n, double alpha, double beta)

Methods inherited from class java.lang.Object

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

Constructor Detail

CudnnLocalResponseNormalizationHelper

public CudnnLocalResponseNormalizationHelper()

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,
                                                                            double k,
                                                                            double n,
                                                                            double alpha,
                                                                            double beta)

Specified by:

backpropGradient in interface LocalResponseNormalizationHelper

activate

public org.nd4j.linalg.api.ndarray.INDArray activate(org.nd4j.linalg.api.ndarray.INDArray input,
                                                     boolean training,
                                                     double k,
                                                     double n,
                                                     double alpha,
                                                     double beta)

Specified by:

activate in interface LocalResponseNormalizationHelper