Package org.deeplearning4j.preprocessors
Class PermutePreprocessor
- java.lang.Object
-
- org.deeplearning4j.nn.conf.preprocessor.BaseInputPreProcessor
-
- org.deeplearning4j.preprocessors.PermutePreprocessor
-
- All Implemented Interfaces:
Serializable
,Cloneable
,InputPreProcessor
public class PermutePreprocessor extends BaseInputPreProcessor
- See Also:
- Serialized Form
-
-
Constructor Summary
Constructors Constructor Description PermutePreprocessor(int... permutationIndices)
-
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.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-
Methods inherited from class org.deeplearning4j.nn.conf.preprocessor.BaseInputPreProcessor
clone, feedForwardMaskArray
-
-
-
-
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- 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.- 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
-
getOutputType
public InputType getOutputType(InputType inputType) throws InvalidInputTypeException
Description copied from interface:InputPreProcessor
For a given type of input to this preprocessor, what is the type of the output?- Parameters:
inputType
- Type of input for the preprocessor- Returns:
- Type of input after applying the preprocessor
- Throws:
InvalidInputTypeException
-
-