org.deeplearning4j.nn.conf.layers

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

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()
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/columns
Deconvolution2D.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

Methods inherited from class java.lang.Object

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

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

dataFormat

public Deconvolution2D.Builder dataFormat(CNN2DFormat format)

allowCausal

protected boolean allowCausal()

Specified by:

allowCausal in class ConvolutionLayer.BaseConvBuilder<Deconvolution2D.Builder>

convolutionMode

public Deconvolution2D.Builder convolutionMode(ConvolutionMode convolutionMode)

Set the convolution mode for the Convolution layer. See ConvolutionMode for more details

Overrides:

convolutionMode in class ConvolutionLayer.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 class ConvolutionLayer.BaseConvBuilder<Deconvolution2D.Builder>

Parameters:

kernelSize - the height and width of the kernel

stride

public Deconvolution2D.Builder stride(int... stride)

Overrides:

stride in class ConvolutionLayer.BaseConvBuilder<Deconvolution2D.Builder>

padding

public Deconvolution2D.Builder padding(int... padding)

Overrides:

padding in class ConvolutionLayer.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 class Layer.Builder<Deconvolution2D.Builder>