org.deeplearning4j.nn.modelimport.keras.preprocessors

Class PermutePreprocessor

java.lang.Object

org.deeplearning4j.nn.conf.preprocessor.BaseInputPreProcessor

org.deeplearning4j.nn.modelimport.keras.preprocessors.PermutePreprocessor

All Implemented Interfaces:

Serializable, Cloneable, InputPreProcessor

public class PermutePreprocessor
extends BaseInputPreProcessor

Preprocessor to permute input data according to specified permutation indices.

Author:

Max Pumperla

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
PermutePreprocessor(int[] permutationIndices)

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

PermutePreprocessor

public PermutePreprocessor(int[] permutationIndices)

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