public class ComposableInputPreProcessor extends BaseInputPreProcessor
Constructor and Description |
---|
ComposableInputPreProcessor(InputPreProcessor... inputPreProcessors) |
Modifier and Type | Method and Description |
---|---|
INDArray |
backprop(INDArray output,
int miniBatchSize,
LayerWorkspaceMgr workspaceMgr)
Reverse the preProcess during backprop.
|
ComposableInputPreProcessor |
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
|
public ComposableInputPreProcessor(InputPreProcessor... inputPreProcessors)
public INDArray preProcess(INDArray input, int miniBatchSize, LayerWorkspaceMgr workspaceMgr)
InputPreProcessor
input
- the input to pre preProcessminiBatchSize
- Minibatch sizeworkspaceMgr
- Workspace managerArrayType.ACTIVATIONS
workspace via the workspace managerpublic INDArray backprop(INDArray output, int miniBatchSize, LayerWorkspaceMgr workspaceMgr)
InputPreProcessor
output
- which is a pair of the gradient and epsilonminiBatchSize
- Minibatch sizeworkspaceMgr
- Workspace managerArrayType.ACTIVATION_GRAD
workspace via the
workspace managerpublic ComposableInputPreProcessor clone()
clone
in interface InputPreProcessor
clone
in class BaseInputPreProcessor
public InputType getOutputType(InputType inputType)
InputPreProcessor
inputType
- Type of input for the preprocessorpublic Pair<INDArray,MaskState> feedForwardMaskArray(INDArray maskArray, MaskState currentMaskState, int minibatchSize)
feedForwardMaskArray
in interface InputPreProcessor
feedForwardMaskArray
in class BaseInputPreProcessor
Copyright © 2019. All rights reserved.