org.deeplearning4j.nn.conf.layers

Class InputTypeUtil

java.lang.Object

org.deeplearning4j.nn.conf.layers.InputTypeUtil

public class InputTypeUtil
extends Object

Utilities for calculating input types

Author:

Alex Black

Constructor Summary

Constructors

Constructor and Description
InputTypeUtil()

Method Summary

Modifier and Type	Method and Description
static InputType	getOutputTypeCnn3DLayers(InputType inputType, int[] kernelSize, int[] stride, int[] padding, int[] dilation, ConvolutionMode convolutionMode, int outputChannels, int layerIdx, String layerName, Class<?> layerClass)
static InputType	getOutputTypeCnnLayers(InputType inputType, int[] kernelSize, int[] stride, int[] padding, int[] dilation, ConvolutionMode convolutionMode, int outputDepth, int layerIdx, String layerName, Class<?> layerClass)
static InputType	getOutputTypeDeconvLayer(InputType inputType, int[] kernelSize, int[] stride, int[] padding, int[] dilation, ConvolutionMode convolutionMode, int outputDepth, int layerIdx, String layerName, Class<?> layerClass)
static InputPreProcessor	getPreProcessorForInputTypeCnn3DLayers(InputType inputType, String layerName) Utility method for determining the appropriate preprocessor for CNN layers, such as ConvolutionLayer and SubsamplingLayer
static InputPreProcessor	getPreProcessorForInputTypeCnnLayers(InputType inputType, String layerName) Utility method for determining the appropriate preprocessor for CNN layers, such as ConvolutionLayer and SubsamplingLayer
static InputPreProcessor	getPreprocessorForInputTypeRnnLayers(InputType inputType, String layerName)

Methods inherited from class java.lang.Object

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

Constructor Detail

InputTypeUtil

public InputTypeUtil()

Method Detail

getOutputTypeDeconvLayer

public static InputType getOutputTypeDeconvLayer(InputType inputType,
                                                 int[] kernelSize,
                                                 int[] stride,
                                                 int[] padding,
                                                 int[] dilation,
                                                 ConvolutionMode convolutionMode,
                                                 int outputDepth,
                                                 int layerIdx,
                                                 String layerName,
                                                 Class<?> layerClass)

getOutputTypeCnn3DLayers

public static InputType getOutputTypeCnn3DLayers(InputType inputType,
                                                 int[] kernelSize,
                                                 int[] stride,
                                                 int[] padding,
                                                 int[] dilation,
                                                 ConvolutionMode convolutionMode,
                                                 int outputChannels,
                                                 int layerIdx,
                                                 String layerName,
                                                 Class<?> layerClass)

getOutputTypeCnnLayers

public static InputType getOutputTypeCnnLayers(InputType inputType,
                                               int[] kernelSize,
                                               int[] stride,
                                               int[] padding,
                                               int[] dilation,
                                               ConvolutionMode convolutionMode,
                                               int outputDepth,
                                               int layerIdx,
                                               String layerName,
                                               Class<?> layerClass)

getPreProcessorForInputTypeCnn3DLayers

public static InputPreProcessor getPreProcessorForInputTypeCnn3DLayers(InputType inputType,
                                                                       String layerName)

Utility method for determining the appropriate preprocessor for CNN layers, such as ConvolutionLayer and SubsamplingLayer

Parameters:

inputType - Input type to get the preprocessor for

Returns:

Null if no preprocessor is required; otherwise the appropriate preprocessor for the given input type

getPreProcessorForInputTypeCnnLayers

public static InputPreProcessor getPreProcessorForInputTypeCnnLayers(InputType inputType,
                                                                     String layerName)

Utility method for determining the appropriate preprocessor for CNN layers, such as ConvolutionLayer and SubsamplingLayer

Parameters:

inputType - Input type to get the preprocessor for

Returns:

Null if no preprocessor is required; otherwise the appropriate preprocessor for the given input type

getPreprocessorForInputTypeRnnLayers

public static InputPreProcessor getPreprocessorForInputTypeRnnLayers(InputType inputType,
                                                                     String layerName)