Class CnnToRnnPreProcessor
- java.lang.Object
-
- org.deeplearning4j.nn.conf.preprocessor.CnnToRnnPreProcessor
-
- All Implemented Interfaces:
Serializable
,Cloneable
,InputPreProcessor
public class CnnToRnnPreProcessor extends Object implements InputPreProcessor
- See Also:
- Serialized Form
-
-
Constructor Summary
Constructors Constructor Description CnnToRnnPreProcessor(long inputHeight, long inputWidth, long numChannels)
CnnToRnnPreProcessor(long inputHeight, long inputWidth, long numChannels, RNNFormat rnnDataFormat)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description INDArray
backprop(INDArray output, int miniBatchSize, LayerWorkspaceMgr workspaceMgr)
Reverse the preProcess during backprop.CnnToRnnPreProcessor
clone()
Pair<INDArray,MaskState>
feedForwardMaskArray(INDArray maskArray, MaskState currentMaskState, int minibatchSize)
InputType
getOutputType(InputType inputType)
For a given type of input to this preprocessor, what is the type of the output?INDArray
preProcess(INDArray input, int miniBatchSize, LayerWorkspaceMgr workspaceMgr)
Pre preProcess input/activations for a multi layer network
-
-
-
Constructor Detail
-
CnnToRnnPreProcessor
public CnnToRnnPreProcessor(long inputHeight, long inputWidth, long numChannels, RNNFormat rnnDataFormat)
-
CnnToRnnPreProcessor
public CnnToRnnPreProcessor(long inputHeight, long inputWidth, long numChannels)
-
-
Method Detail
-
preProcess
public INDArray preProcess(INDArray input, int miniBatchSize, LayerWorkspaceMgr workspaceMgr)
Description copied from interface:InputPreProcessor
Pre preProcess input/activations for a multi layer network- Specified by:
preProcess
in interfaceInputPreProcessor
- Parameters:
input
- the input to pre preProcessminiBatchSize
- Minibatch sizeworkspaceMgr
- Workspace manager- Returns:
- the processed input. Note that the returned array should be placed in the
ArrayType.ACTIVATIONS
workspace via the workspace manager
-
backprop
public INDArray backprop(INDArray output, int miniBatchSize, LayerWorkspaceMgr workspaceMgr)
Description copied from interface:InputPreProcessor
Reverse the preProcess during backprop. Process Gradient/epsilons before passing them to the layer below.- Specified by:
backprop
in interfaceInputPreProcessor
- Parameters:
output
- which is a pair of the gradient and epsilonminiBatchSize
- Minibatch sizeworkspaceMgr
- Workspace manager- Returns:
- the reverse of the pre preProcess step (if any). Note that the returned array should be
placed in
ArrayType.ACTIVATION_GRAD
workspace via the workspace manager
-
clone
public CnnToRnnPreProcessor clone()
- Specified by:
clone
in interfaceInputPreProcessor
- Overrides:
clone
in classObject
-
getOutputType
public InputType getOutputType(InputType inputType)
Description copied from interface:InputPreProcessor
For a given type of input to this preprocessor, what is the type of the output?- Specified by:
getOutputType
in interfaceInputPreProcessor
- Parameters:
inputType
- Type of input for the preprocessor- Returns:
- Type of input after applying the preprocessor
-
feedForwardMaskArray
public Pair<INDArray,MaskState> feedForwardMaskArray(INDArray maskArray, MaskState currentMaskState, int minibatchSize)
- Specified by:
feedForwardMaskArray
in interfaceInputPreProcessor
-
-