org.deeplearning4j.nn.conf.inputs

Class InputType

java.lang.Object

org.deeplearning4j.nn.conf.inputs.InputType

All Implemented Interfaces:

Serializable

Direct Known Subclasses:

InputType.InputTypeConvolutional, InputType.InputTypeConvolutional3D, InputType.InputTypeConvolutionalFlat, InputType.InputTypeFeedForward, InputType.InputTypeRecurrent

public abstract class InputType
extends Object
implements Serializable

The InputType class is used to track and define the types of activations etc used in a ComputationGraph. This is most useful for automatically adding preprocessors between layers, and automatically setting nIn values. See: ComputationGraphConfiguration.GraphBuilder.setInputTypes(InputType...) and ComputationGraphConfiguration.addPreProcessors(InputType...)

Author:

Alex Black

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	InputType.InputTypeConvolutional
static class	InputType.InputTypeConvolutional3D
static class	InputType.InputTypeConvolutionalFlat
static class	InputType.InputTypeFeedForward
static class	InputType.InputTypeRecurrent
static class	InputType.Type The type of activations in/out of a given GraphVertex FF: Standard feed-foward (2d minibatch, 1d per example) data RNN: Recurrent neural network (3d minibatch) time series data CNN: 2D Convolutional neural network (4d minibatch, [miniBatchSize, channels, height, width]) CNNFlat: Flattened 2D conv net data (2d minibatch, [miniBatchSize, height * width * channels]) CNN3D: 3D convolutional neural network (5d minibatch, [miniBatchSize, channels, height, width, channels])

Constructor Summary

Constructors

Constructor and Description
InputType()

Method Summary

Modifier and Type	Method and Description
abstract int	arrayElementsPerExample()
static InputType	convolutional(int height, int width, int depth) Input type for convolutional (CNN) data, that is 4d with shape [miniBatchSize, channels, height, width].
static InputType	convolutional3D(int depth, int height, int width, int channels) Input type for 3D convolutional (CNN3D) data, that is 5d with shape [miniBatchSize, channels, height, width, channels].
static InputType	convolutionalFlat(int height, int width, int depth) Input type for convolutional (CNN) data, where the data is in flattened (row vector) format.
static InputType	feedForward(int size) InputType for feed forward network data
int[]	getShape() Returns the shape of this InputType without minibatch dimension in the returned array
abstract int[]	getShape(boolean includeBatchDim) Returns the shape of this InputType
abstract InputType.Type	getType()
static InputType	inferInputType(org.nd4j.linalg.api.ndarray.INDArray inputArray)
static InputType[]	inferInputTypes(org.nd4j.linalg.api.ndarray.INDArray... inputArrays)
static InputType	recurrent(int size) InputType for recurrent neural network (time series) data
static InputType	recurrent(int size, int timeSeriesLength) InputType for recurrent neural network (time series) data
abstract String	toString()

Methods inherited from class java.lang.Object

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

Constructor Detail

InputType

public InputType()

Method Detail

getType

public abstract InputType.Type getType()

toString

public abstract String toString()

Overrides:

toString in class Object

arrayElementsPerExample

public abstract int arrayElementsPerExample()

getShape

public abstract int[] getShape(boolean includeBatchDim)

Returns the shape of this InputType

Parameters:

includeBatchDim - Whether to include minibatch in the return shape array

Returns:

int[]

getShape

public int[] getShape()

Returns the shape of this InputType without minibatch dimension in the returned array

Returns:

int[]

feedForward

public static InputType feedForward(int size)

InputType for feed forward network data

Parameters:

size - The size of the activations

Returns:

InputTypeFeedForward

recurrent

public static InputType recurrent(int size)

InputType for recurrent neural network (time series) data

Parameters:

size - The size of the activations

Returns:

InputTypeRecurrent

recurrent

public static InputType recurrent(int size,
                                  int timeSeriesLength)

InputType for recurrent neural network (time series) data

Parameters:

size - The size of the activations

timeSeriesLength - Length of the input time series

Returns:

InputTypeRecurrent

convolutional

public static InputType convolutional(int height,
                                      int width,
                                      int depth)

Input type for convolutional (CNN) data, that is 4d with shape [miniBatchSize, channels, height, width]. For CNN data that has been flattened, use convolutionalFlat(int, int, int)

Parameters:

height - height of the input

width - Width of the input

depth - Depth, or number of channels

Returns:

InputTypeConvolutional

convolutional3D

public static InputType convolutional3D(int depth,
                                        int height,
                                        int width,
                                        int channels)

Input type for 3D convolutional (CNN3D) data, that is 5d with shape [miniBatchSize, channels, height, width, channels].

Parameters:

height - height of the input

width - Width of the input

depth - Depth of the input

channels - Number of channels of the input

Returns:

InputTypeConvolutional3D

convolutionalFlat

public static InputType convolutionalFlat(int height,
                                          int width,
                                          int depth)

Input type for convolutional (CNN) data, where the data is in flattened (row vector) format. Expect data with shape [miniBatchSize, height * width * channels]. For CNN data in 4d format, use convolutional(int, int, int)

Parameters:

height - Height of the (unflattened) data represented by this input type

width - Width of the (unflattened) data represented by this input type

depth - Depth of the (unflattened) data represented by this input type

Returns:

InputTypeConvolutionalFlat

inferInputType

public static InputType inferInputType(org.nd4j.linalg.api.ndarray.INDArray inputArray)

inferInputTypes

public static InputType[] inferInputTypes(org.nd4j.linalg.api.ndarray.INDArray... inputArrays)