Class Deconvolution2D.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<Deconvolution2D.Builder>
-
- org.deeplearning4j.nn.conf.layers.Deconvolution2D.Builder
-
- Enclosing class:
- Deconvolution2D
public static class Deconvolution2D.Builder extends ConvolutionLayer.BaseConvBuilder<Deconvolution2D.Builder>
-
-
Field Summary
-
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()
Deconvolution2D
build()
Deconvolution2D.Builder
convolutionMode(ConvolutionMode convolutionMode)
Set the convolution mode for the Convolution layer.Deconvolution2D.Builder
dataFormat(CNN2DFormat format)
Deconvolution2D.Builder
kernelSize(int... kernelSize)
Size of the convolution rows/columnsDeconvolution2D.Builder
padding(int... padding)
void
setDilation(int... dilation)
void
setKernelSize(int... kernelSize)
void
setPadding(int... padding)
void
setStride(int... stride)
Deconvolution2D.Builder
stride(int... stride)
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.ConvolutionLayer.BaseConvBuilder
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
-
-
-
-
Method Detail
-
dataFormat
public Deconvolution2D.Builder dataFormat(CNN2DFormat format)
-
allowCausal
protected boolean allowCausal()
- Specified by:
allowCausal
in classConvolutionLayer.BaseConvBuilder<Deconvolution2D.Builder>
-
convolutionMode
public Deconvolution2D.Builder convolutionMode(ConvolutionMode convolutionMode)
Set the convolution mode for the Convolution layer. SeeConvolutionMode
for more details- Overrides:
convolutionMode
in classConvolutionLayer.BaseConvBuilder<Deconvolution2D.Builder>
- Parameters:
convolutionMode
- Convolution mode for layer
-
kernelSize
public Deconvolution2D.Builder kernelSize(int... kernelSize)
Size of the convolution rows/columns- Overrides:
kernelSize
in classConvolutionLayer.BaseConvBuilder<Deconvolution2D.Builder>
- Parameters:
kernelSize
- the height and width of the kernel
-
stride
public Deconvolution2D.Builder stride(int... stride)
- Overrides:
stride
in classConvolutionLayer.BaseConvBuilder<Deconvolution2D.Builder>
-
padding
public Deconvolution2D.Builder padding(int... padding)
- Overrides:
padding
in classConvolutionLayer.BaseConvBuilder<Deconvolution2D.Builder>
-
setKernelSize
public void setKernelSize(int... kernelSize)
-
setStride
public void setStride(int... stride)
-
setPadding
public void setPadding(int... padding)
-
setDilation
public void setDilation(int... dilation)
-
build
public Deconvolution2D build()
- Specified by:
build
in classLayer.Builder<Deconvolution2D.Builder>
-
-