org.deeplearning4j.nn.conf.layers

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

Constructor Summary

Constructors

Constructor and Description
Builder()
Builder(int kernelSize) Constructor with specified kernel size, stride of 1, padding of 0
Builder(int kernelSize, int stride)
Builder(int kernelSize, int stride, int padding)

Method Summary

Modifier and Type	Method and Description
protected boolean	allowCausal()
Convolution1DLayer	build()
Convolution1DLayer.Builder	kernelSize(int kernelSize) Size of the convolution
Convolution1DLayer.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

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Builder

public Builder(int kernelSize,
               int stride)

Parameters:

kernelSize - Kernel size

stride - 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 size

stride - Stride

padding - Padding

Method Detail

allowCausal

protected boolean allowCausal()

Specified by:

allowCausal in class ConvolutionLayer.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 with ConvolutionMode.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 class Layer.Builder<Convolution1DLayer.Builder>