public interface InputPreProcessor
extends java.io.Serializable, java.lang.Cloneable
Modifier and Type | Method and Description |
---|---|
org.nd4j.linalg.api.ndarray.INDArray |
backprop(org.nd4j.linalg.api.ndarray.INDArray output,
int miniBatchSize)
Reverse the preProcess during backprop.
|
InputPreProcessor |
clone() |
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)
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)
input
- the input to pre preProcessminiBatchSize
- org.nd4j.linalg.api.ndarray.INDArray backprop(org.nd4j.linalg.api.ndarray.INDArray output, int miniBatchSize)
output
- which is a pair of the gradient and epsilonminiBatchSize
- InputPreProcessor clone()
InputType getOutputType(InputType inputType)
inputType
- Type of input for the preprocessor