Class Deconvolution3D.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<Deconvolution3D.Builder>
-
- org.deeplearning4j.nn.conf.layers.Deconvolution3D.Builder
-
- Enclosing class:
- Deconvolution3D
public static class Deconvolution3D.Builder extends ConvolutionLayer.BaseConvBuilder<Deconvolution3D.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
-
-
Constructor Summary
Constructors Constructor Description Builder()
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description protected boolean
allowCausal()
Deconvolution3D
build()
Deconvolution3D.Builder
convolutionMode(ConvolutionMode convolutionMode)
Set the convolution mode for the Convolution layer.Deconvolution3D.Builder
dataFormat(Convolution3D.DataFormat dataFormat)
Deconvolution3D.Builder
kernelSize(int... kernelSize)
Size of the convolution rows/columnsDeconvolution3D.Builder
padding(int... padding)
void
setDilation(int... dilation)
void
setKernelSize(int... kernelSize)
void
setPadding(int... padding)
void
setStride(int... stride)
Deconvolution3D.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
-
allowCausal
protected boolean allowCausal()
- Specified by:
allowCausal
in classConvolutionLayer.BaseConvBuilder<Deconvolution3D.Builder>
-
convolutionMode
public Deconvolution3D.Builder convolutionMode(ConvolutionMode convolutionMode)
Set the convolution mode for the Convolution layer. SeeConvolutionMode
for more details- Overrides:
convolutionMode
in classConvolutionLayer.BaseConvBuilder<Deconvolution3D.Builder>
- Parameters:
convolutionMode
- Convolution mode for layer
-
kernelSize
public Deconvolution3D.Builder kernelSize(int... kernelSize)
Size of the convolution rows/columns- Overrides:
kernelSize
in classConvolutionLayer.BaseConvBuilder<Deconvolution3D.Builder>
- Parameters:
kernelSize
- the height and width of the kernel
-
stride
public Deconvolution3D.Builder stride(int... stride)
- Overrides:
stride
in classConvolutionLayer.BaseConvBuilder<Deconvolution3D.Builder>
-
padding
public Deconvolution3D.Builder padding(int... padding)
- Overrides:
padding
in classConvolutionLayer.BaseConvBuilder<Deconvolution3D.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)
-
dataFormat
public Deconvolution3D.Builder dataFormat(Convolution3D.DataFormat dataFormat)
-
build
public Deconvolution3D build()
- Specified by:
build
in classLayer.Builder<Deconvolution3D.Builder>
-
-