org.deeplearning4j.nn.conf.layers

Class ConvolutionLayer.Builder

java.lang.Object

org.deeplearning4j.nn.conf.layers.Layer.Builder<T>

org.deeplearning4j.nn.conf.layers.BaseLayer.Builder<T>

org.deeplearning4j.nn.conf.layers.FeedForwardLayer.Builder<T>

org.deeplearning4j.nn.conf.layers.ConvolutionLayer.BaseConvBuilder<ConvolutionLayer.Builder>

org.deeplearning4j.nn.conf.layers.ConvolutionLayer.Builder

Enclosing class:

ConvolutionLayer

public static class ConvolutionLayer.Builder
extends ConvolutionLayer.BaseConvBuilder<ConvolutionLayer.Builder>

Field Summary

Fields

Modifier and Type	Field and Description
protected CNN2DFormat	dataFormat

Fields inherited from class org.deeplearning4j.nn.conf.layers.ConvolutionLayer.BaseConvBuilder

convolutionDim, convolutionMode, cudnnAlgoMode, cudnnAllowFallback, cudnnBwdDataAlgo, cudnnBwdFilterAlgo, cudnnFwdAlgo, dilation, hasBias, kernelSize, padding, stride

Fields inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer.Builder

nIn, nOut

Fields inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer.Builder

activationFn, biasInit, biasUpdater, gainInit, gradientNormalization, gradientNormalizationThreshold, iupdater, regularization, regularizationBias, weightInitFn, weightNoise

Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder

allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints

Constructor Summary

Constructors

Constructor and Description
Builder()
Builder(int... kernelSize)
Builder(int[] kernelSize, int[] stride)
Builder(int[] kernelSize, int[] stride, int[] padding)

Method Summary

Modifier and Type	Method and Description
protected boolean	allowCausal()
ConvolutionLayer	build()
ConvolutionLayer.Builder	dataFormat(CNN2DFormat format) Set the data format for the CNN activations - NCHW (channels first) or NHWC (channels last).
ConvolutionLayer.Builder	kernelSize(int... kernelSize) Size of the convolution rows/columns
ConvolutionLayer.Builder	padding(int... padding)
void	setDataFormat(CNN2DFormat dataFormat)
void	setDilation(int... dilation) Set dilation size for 3D convolutions in (depth, height, width) order
void	setKernelSize(int... kernelSize) Set kernel size for 3D convolutions in (depth, height, width) order
void	setPadding(int... padding) Set padding size for 3D convolutions in (depth, height, width) order
void	setStride(int... stride) Set stride size for 3D convolutions in (depth, height, width) order
ConvolutionLayer.Builder	stride(int... stride)

Methods inherited from class org.deeplearning4j.nn.conf.layers.ConvolutionLayer.BaseConvBuilder

convolutionMode, cudnnAlgoMode, cudnnAllowFallback, cudnnBwdDataMode, cudnnBwdFilterMode, cudnnFwdMode, dilation, hasBias, helperAllowFallback, setConvolutionMode

Methods inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer.Builder

nIn, nIn, nOut, nOut, units

Methods inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer.Builder

activation, activation, biasInit, biasUpdater, dist, gainInit, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, regularization, regularizationBias, updater, updater, weightDecay, weightDecay, weightDecayBias, weightDecayBias, weightInit, weightInit, weightInit, weightNoise

Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder

constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name

Methods inherited from class java.lang.Object

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

Field Detail

dataFormat

protected CNN2DFormat dataFormat

Constructor Detail

Builder

public Builder(int[] kernelSize,
               int[] stride,
               int[] padding)

Builder

public Builder(int[] kernelSize,
               int[] stride)

Builder

public Builder(int... kernelSize)

Builder

public Builder()

Method Detail

allowCausal

protected boolean allowCausal()

Specified by:

allowCausal in class ConvolutionLayer.BaseConvBuilder<ConvolutionLayer.Builder>

kernelSize

public ConvolutionLayer.Builder kernelSize(int... kernelSize)

Size of the convolution rows/columns

Overrides:

kernelSize in class ConvolutionLayer.BaseConvBuilder<ConvolutionLayer.Builder>

Parameters:

kernelSize - the height and width of the kernel

stride

public ConvolutionLayer.Builder stride(int... stride)

Overrides:

stride in class ConvolutionLayer.BaseConvBuilder<ConvolutionLayer.Builder>

padding

public ConvolutionLayer.Builder padding(int... padding)

Overrides:

padding in class ConvolutionLayer.BaseConvBuilder<ConvolutionLayer.Builder>

dataFormat

public ConvolutionLayer.Builder dataFormat(CNN2DFormat format)

Set the data format for the CNN activations - NCHW (channels first) or NHWC (channels last). See CNN2DFormat for more details.
Default: NCHW

Parameters:

format - Format for activations (in and out)

build

public ConvolutionLayer build()

Specified by:

build in class Layer.Builder<ConvolutionLayer.Builder>

setKernelSize

public void setKernelSize(int... kernelSize)

Set kernel size for 3D convolutions in (depth, height, width) order

Parameters:

kernelSize - kernel size

setStride

public void setStride(int... stride)

Set stride size for 3D convolutions in (depth, height, width) order

Parameters:

stride - kernel size

setPadding

public void setPadding(int... padding)

Set padding size for 3D convolutions in (depth, height, width) order

Parameters:

padding - kernel size

setDilation

public void setDilation(int... dilation)

Set dilation size for 3D convolutions in (depth, height, width) order

Parameters:

dilation - kernel size

setDataFormat

public void setDataFormat(CNN2DFormat dataFormat)