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 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
-
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method 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/columnsConvolutionLayer.Builder
padding(int... padding)
void
setDataFormat(CNN2DFormat dataFormat)
void
setDilation(int... dilation)
Set dilation size for 3D convolutions in (depth, height, width) ordervoid
setKernelSize(int... kernelSize)
Set kernel size for 3D convolutions in (depth, height, width) ordervoid
setPadding(int... padding)
Set padding size for 3D convolutions in (depth, height, width) ordervoid
setStride(int... stride)
Set stride size for 3D convolutions in (depth, height, width) orderConvolutionLayer.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
-
-
-
-
Field Detail
-
dataFormat
protected CNN2DFormat dataFormat
-
-
Method Detail
-
allowCausal
protected boolean allowCausal()
- Specified by:
allowCausal
in classConvolutionLayer.BaseConvBuilder<ConvolutionLayer.Builder>
-
kernelSize
public ConvolutionLayer.Builder kernelSize(int... kernelSize)
Size of the convolution rows/columns- Overrides:
kernelSize
in classConvolutionLayer.BaseConvBuilder<ConvolutionLayer.Builder>
- Parameters:
kernelSize
- the height and width of the kernel
-
stride
public ConvolutionLayer.Builder stride(int... stride)
- Overrides:
stride
in classConvolutionLayer.BaseConvBuilder<ConvolutionLayer.Builder>
-
padding
public ConvolutionLayer.Builder padding(int... padding)
- Overrides:
padding
in classConvolutionLayer.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). SeeCNN2DFormat
for more details.
Default: NCHW- Parameters:
format
- Format for activations (in and out)
-
build
public ConvolutionLayer build()
- Specified by:
build
in classLayer.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)
-
-