public interface InputPreProcessor extends Serializable, Cloneable
Modifier and Type | Method and Description |
---|---|
org.nd4j.linalg.api.ndarray.INDArray |
backprop(org.nd4j.linalg.api.ndarray.INDArray output,
int miniBatchSize,
LayerWorkspaceMgr workspaceMgr)
Reverse the preProcess during backprop.
|
InputPreProcessor |
clone() |
org.nd4j.linalg.primitives.Pair<org.nd4j.linalg.api.ndarray.INDArray,MaskState> |
feedForwardMaskArray(org.nd4j.linalg.api.ndarray.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?
|
org.nd4j.linalg.api.ndarray.INDArray |
preProcess(org.nd4j.linalg.api.ndarray.INDArray input,
int miniBatchSize,
LayerWorkspaceMgr workspaceMgr)
Pre preProcess input/activations for a multi layer network
|
org.nd4j.linalg.api.ndarray.INDArray preProcess(org.nd4j.linalg.api.ndarray.INDArray input, int miniBatchSize, LayerWorkspaceMgr workspaceMgr)
input
- the input to pre preProcessminiBatchSize
- Minibatch sizeworkspaceMgr
- Workspace managerArrayType.ACTIVATIONS
workspace via the workspace managerorg.nd4j.linalg.api.ndarray.INDArray backprop(org.nd4j.linalg.api.ndarray.INDArray output, int miniBatchSize, LayerWorkspaceMgr workspaceMgr)
output
- which is a pair of the gradient and epsilonminiBatchSize
- Minibatch sizeworkspaceMgr
- Workspace managerArrayType.ACTIVATION_GRAD
workspace via the
workspace managerInputPreProcessor clone()
InputType getOutputType(InputType inputType)
inputType
- Type of input for the preprocessorCopyright © 2018. All rights reserved.