org.deeplearning4j.nn.layers.dropout

Class CudnnDropoutHelper

java.lang.Object

org.deeplearning4j.nn.layers.BaseCudnnHelper

org.deeplearning4j.nn.layers.dropout.CudnnDropoutHelper

All Implemented Interfaces:

org.deeplearning4j.nn.conf.dropout.DropoutHelper

public class CudnnDropoutHelper
extends BaseCudnnHelper
implements org.deeplearning4j.nn.conf.dropout.DropoutHelper

CuDNN dropout helper Note that for repeatability between calls (for example, for gradient checks), we need to do two things: (a) set the ND4J RNG seed (b) clear the rngStates field

Author:

Alex Black

Nested Class Summary

Nested classes/interfaces inherited from class org.deeplearning4j.nn.layers.BaseCudnnHelper

BaseCudnnHelper.CudnnContext, BaseCudnnHelper.DataCache, BaseCudnnHelper.TensorArray

Field Summary

Fields inherited from class org.deeplearning4j.nn.layers.BaseCudnnHelper

alpha, beta, dataType, dataTypeSize, sizeInBytes, TENSOR_FORMAT

Constructor Summary

Constructors

Constructor and Description
CudnnDropoutHelper()

Method Summary

Modifier and Type	Method and Description
void	applyDropout(INDArray input, INDArray resultArray, double dropoutInputRetainProb)
void	backprop(INDArray gradAtOutput, INDArray gradAtInput)

Methods inherited from class org.deeplearning4j.nn.layers.BaseCudnnHelper

adaptForTensorDescr, checkCuda, checkCudnn, checkSupported, toCudnnDataType

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.deeplearning4j.nn.conf.dropout.DropoutHelper

checkSupported

Constructor Detail

CudnnDropoutHelper

public CudnnDropoutHelper()

Method Detail

applyDropout

public void applyDropout(INDArray input,
                         INDArray resultArray,
                         double dropoutInputRetainProb)

Specified by:

applyDropout in interface org.deeplearning4j.nn.conf.dropout.DropoutHelper

backprop

public void backprop(INDArray gradAtOutput,
                     INDArray gradAtInput)

Specified by:

backprop in interface org.deeplearning4j.nn.conf.dropout.DropoutHelper