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)
InputPreProcessorinput - the input to pre preProcessminiBatchSize - Minibatch sizeworkspaceMgr - Workspace managerArrayType.ACTIVATIONS workspace via the workspace managerpublic INDArray backprop(INDArray output, int miniBatchSize, LayerWorkspaceMgr workspaceMgr)
InputPreProcessoroutput - 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 InputPreProcessorclone in class BaseInputPreProcessorpublic InputType getOutputType(InputType inputType)
InputPreProcessorinputType - Type of input for the preprocessorpublic Pair<INDArray,MaskState> feedForwardMaskArray(INDArray maskArray, MaskState currentMaskState, int minibatchSize)
feedForwardMaskArray in interface InputPreProcessorfeedForwardMaskArray in class BaseInputPreProcessorCopyright © 2020. All rights reserved.