public class CnnToFeedForwardPreProcessor extends Object implements InputPreProcessor
for opposite case (i.e., DenseLayer -> CNNetc)
,
Serialized FormModifier and Type | Field and Description |
---|---|
protected long |
inputHeight |
protected long |
inputWidth |
protected long |
numChannels |
Constructor and Description |
---|
CnnToFeedForwardPreProcessor() |
CnnToFeedForwardPreProcessor(long inputHeight,
long inputWidth) |
CnnToFeedForwardPreProcessor(long inputHeight,
long inputWidth,
long numChannels) |
Modifier and Type | Method and Description |
---|---|
INDArray |
backprop(INDArray epsilons,
int miniBatchSize,
LayerWorkspaceMgr workspaceMgr)
Reverse the preProcess during backprop.
|
CnnToFeedForwardPreProcessor |
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
|
protected long inputHeight
protected long inputWidth
protected long numChannels
public CnnToFeedForwardPreProcessor(long inputHeight, long inputWidth, long numChannels)
inputHeight
- the columnsinputWidth
- the rowsnumChannels
- the channelspublic CnnToFeedForwardPreProcessor(long inputHeight, long inputWidth)
public CnnToFeedForwardPreProcessor()
public INDArray preProcess(INDArray input, int miniBatchSize, LayerWorkspaceMgr workspaceMgr)
InputPreProcessor
preProcess
in interface InputPreProcessor
input
- the input to pre preProcessminiBatchSize
- Minibatch sizeworkspaceMgr
- Workspace managerArrayType.ACTIVATIONS
workspace via the workspace managerpublic INDArray backprop(INDArray epsilons, int miniBatchSize, LayerWorkspaceMgr workspaceMgr)
InputPreProcessor
backprop
in interface InputPreProcessor
epsilons
- which is a pair of the gradient and epsilonminiBatchSize
- Minibatch sizeworkspaceMgr
- Workspace managerArrayType.ACTIVATION_GRAD
workspace via the
workspace managerpublic CnnToFeedForwardPreProcessor clone()
clone
in interface InputPreProcessor
clone
in class Object
public InputType getOutputType(InputType inputType)
InputPreProcessor
getOutputType
in interface InputPreProcessor
inputType
- Type of input for the preprocessorpublic Pair<INDArray,MaskState> feedForwardMaskArray(INDArray maskArray, MaskState currentMaskState, int minibatchSize)
feedForwardMaskArray
in interface InputPreProcessor
Copyright © 2018. All rights reserved.