org.deeplearning4j.nn.modelimport.keras.preprocessors

Class ReshapePreprocessor

java.lang.Object

org.deeplearning4j.nn.conf.preprocessor.BaseInputPreProcessor

org.deeplearning4j.nn.modelimport.keras.preprocessors.ReshapePreprocessor

All Implemented Interfaces:

Serializable, Cloneable, InputPreProcessor

public class ReshapePreprocessor
extends BaseInputPreProcessor

Generic reshape preprocessor

Author:

Max Pumperla

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
ReshapePreprocessor(long[] inputShape, long[] targetShape)

Method Summary

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

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

clone, feedForwardMaskArray

Methods inherited from class java.lang.Object

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

Constructor Detail

ReshapePreprocessor

public ReshapePreprocessor(long[] inputShape,
                           long[] targetShape)

Method Detail

preProcess

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

backprop

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

getOutputType

public InputType getOutputType(InputType inputType)
                        throws InvalidInputTypeException

Throws:

InvalidInputTypeException