org.deeplearning4j.nn.modelimport.keras.preprocessors

Class KerasFlattenRnnPreprocessor

java.lang.Object

org.deeplearning4j.nn.conf.preprocessor.BaseInputPreProcessor

org.deeplearning4j.nn.modelimport.keras.preprocessors.KerasFlattenRnnPreprocessor

All Implemented Interfaces:

Serializable, Cloneable, InputPreProcessor

public class KerasFlattenRnnPreprocessor
extends BaseInputPreProcessor

Preprocessor to flatten input of RNN type

Author:

Max Pumperla

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
KerasFlattenRnnPreprocessor(long depth, long tsLength)

Method Summary

Modifier and Type	Method and Description
INDArray	backprop(INDArray epsilons, int miniBatchSize, LayerWorkspaceMgr workspaceMgr)
KerasFlattenRnnPreprocessor	clone()
InputType	getOutputType(InputType inputType)
INDArray	preProcess(INDArray input, int miniBatchSize, LayerWorkspaceMgr workspaceMgr)

Methods inherited from class org.deeplearning4j.nn.conf.preprocessor.BaseInputPreProcessor

feedForwardMaskArray

Methods inherited from class java.lang.Object

equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

KerasFlattenRnnPreprocessor

public KerasFlattenRnnPreprocessor(long depth,
                                   long tsLength)

Method Detail

preProcess

public INDArray preProcess(INDArray input,
                           int miniBatchSize,
                           LayerWorkspaceMgr workspaceMgr)

backprop

public INDArray backprop(INDArray epsilons,
                         int miniBatchSize,
                         LayerWorkspaceMgr workspaceMgr)

clone

public KerasFlattenRnnPreprocessor clone()

Specified by:

clone in interface InputPreProcessor

Overrides:

clone in class BaseInputPreProcessor

getOutputType

public InputType getOutputType(InputType inputType)
                        throws InvalidInputTypeException

Throws:

InvalidInputTypeException