Class Convolution1DLayer.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<Convolution1DLayer.Builder>
-
- org.deeplearning4j.nn.conf.layers.Convolution1DLayer.Builder
-
- Enclosing class:
- Convolution1DLayer
public static class Convolution1DLayer.Builder extends ConvolutionLayer.BaseConvBuilder<Convolution1DLayer.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()
Convolution1DLayer
build()
Convolution1DLayer.Builder
kernelSize(int kernelSize)
Size of the convolutionConvolution1DLayer.Builder
padding(int padding)
Padding value for the convolution.Convolution1DLayer.Builder
rnnDataFormat(RNNFormat rnnDataFormat)
void
setDilation(int... dilation)
void
setKernelSize(int... kernelSize)
void
setPadding(int... padding)
void
setStride(int... stride)
Convolution1DLayer.Builder
stride(int stride)
Stride for the convolution.-
Methods inherited from class org.deeplearning4j.nn.conf.layers.ConvolutionLayer.BaseConvBuilder
convolutionMode, cudnnAlgoMode, cudnnAllowFallback, cudnnBwdDataMode, cudnnBwdFilterMode, cudnnFwdMode, dilation, hasBias, helperAllowFallback, kernelSize, padding, setConvolutionMode, stride
-
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
-
-
-
-
Constructor Detail
-
Builder
public Builder()
-
Builder
public Builder(int kernelSize, int stride)
- Parameters:
kernelSize
- Kernel sizestride
- Stride
-
Builder
public Builder(int kernelSize)
Constructor with specified kernel size, stride of 1, padding of 0- Parameters:
kernelSize
- Kernel size
-
Builder
public Builder(int kernelSize, int stride, int padding)
- Parameters:
kernelSize
- Kernel sizestride
- Stridepadding
- Padding
-
-
Method Detail
-
allowCausal
protected boolean allowCausal()
- Specified by:
allowCausal
in classConvolutionLayer.BaseConvBuilder<Convolution1DLayer.Builder>
-
rnnDataFormat
public Convolution1DLayer.Builder rnnDataFormat(RNNFormat rnnDataFormat)
-
kernelSize
public Convolution1DLayer.Builder kernelSize(int kernelSize)
Size of the convolution- Parameters:
kernelSize
- the length of the kernel
-
stride
public Convolution1DLayer.Builder stride(int stride)
Stride for the convolution. Must be > 0- Parameters:
stride
- Stride
-
padding
public Convolution1DLayer.Builder padding(int padding)
Padding value for the convolution. Not used withConvolutionMode.Same
- Parameters:
padding
- Padding value
-
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 Convolution1DLayer build()
- Specified by:
build
in classLayer.Builder<Convolution1DLayer.Builder>
-
-