Class LocallyConnected2D.Builder
- java.lang.Object
-
- org.deeplearning4j.nn.conf.layers.Layer.Builder<T>
-
- org.deeplearning4j.nn.conf.layers.samediff.AbstractSameDiffLayer.Builder<T>
-
- org.deeplearning4j.nn.conf.layers.samediff.SameDiffLayer.Builder<LocallyConnected2D.Builder>
-
- org.deeplearning4j.nn.conf.layers.LocallyConnected2D.Builder
-
- Enclosing class:
- LocallyConnected2D
public static class LocallyConnected2D.Builder extends SameDiffLayer.Builder<LocallyConnected2D.Builder>
-
-
Field Summary
Fields Modifier and Type Field Description protected CNN2DFormat
format
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.samediff.SameDiffLayer.Builder
paramWeightInit, weightInit
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.samediff.AbstractSameDiffLayer.Builder
biasUpdater, regularization, regularizationBias, updater
-
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 LocallyConnected2D.Builder
activation(Activation activation)
LocallyConnected2D
build()
LocallyConnected2D.Builder
convolutionMode(ConvolutionMode cm)
LocallyConnected2D.Builder
dataFormat(CNN2DFormat format)
Set the data format for the CNN activations - NCHW (channels first) or NHWC (channels last).LocallyConnected2D.Builder
dilation(int... d)
LocallyConnected2D.Builder
hasBias(boolean hasBias)
LocallyConnected2D.Builder
kernelSize(int... k)
LocallyConnected2D.Builder
nIn(int nIn)
LocallyConnected2D.Builder
nOut(int nOut)
LocallyConnected2D.Builder
padding(int... p)
void
setDilation(int... dilation)
LocallyConnected2D.Builder
setInputSize(int... inputSize)
Set input filter size (h,w) for this locally connected 2D layervoid
setKernel(int... kernel)
void
setPadding(int... padding)
void
setStride(int... stride)
LocallyConnected2D.Builder
stride(int... s)
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.samediff.SameDiffLayer.Builder
weightInit, weightInit
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.samediff.AbstractSameDiffLayer.Builder
biasUpdater, l1, l1Bias, l2, l2Bias, regularization, regularizationBias, updater, weightDecay, weightDecay, weightDecayBias, weightDecayBias
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer.Builder
constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name
-
-
-
-
Field Detail
-
format
protected CNN2DFormat format
-
-
Method Detail
-
setKernel
public void setKernel(int... kernel)
- Parameters:
kernel
- Kernel size for the layer. Must be 2 values (height/width)
-
setStride
public void setStride(int... stride)
- Parameters:
stride
- Stride for the layer. Must be 2 values (height/width)
-
setPadding
public void setPadding(int... padding)
- Parameters:
padding
- Padding for the layer. Not used ifConvolutionMode.Same
is set. Must be 2 values (height/width)
-
setDilation
public void setDilation(int... dilation)
- Parameters:
dilation
- Dilation for the layer. Must be 2 values (height/width)
-
nIn
public LocallyConnected2D.Builder nIn(int nIn)
- Parameters:
nIn
- Number of inputs to the layer (input size)
-
nOut
public LocallyConnected2D.Builder nOut(int nOut)
- Parameters:
nOut
- Number of outputs (output size)
-
activation
public LocallyConnected2D.Builder activation(Activation activation)
- Parameters:
activation
- Activation function for the layer
-
kernelSize
public LocallyConnected2D.Builder kernelSize(int... k)
- Parameters:
k
- Kernel size for the layer. Must be 2 values (height/width)
-
stride
public LocallyConnected2D.Builder stride(int... s)
- Parameters:
s
- Stride for the layer. Must be 2 values (height/width)
-
padding
public LocallyConnected2D.Builder padding(int... p)
- Parameters:
p
- Padding for the layer. Not used ifConvolutionMode.Same
is set. Must be 2 values (height/width)
-
convolutionMode
public LocallyConnected2D.Builder convolutionMode(ConvolutionMode cm)
- Parameters:
cm
- Convolution mode for the layer. SeeConvolutionMode
for details
-
dilation
public LocallyConnected2D.Builder dilation(int... d)
- Parameters:
d
- Dilation for the layer. Must be 2 values (height/width)
-
dataFormat
public LocallyConnected2D.Builder dataFormat(CNN2DFormat format)
Set the data format for the CNN activations - NCHW (channels first) or NHWC (channels last). SeeCNN2DFormat
for more details.
Default: NCHW- Parameters:
format
- Format for activations (in and out)
-
hasBias
public LocallyConnected2D.Builder hasBias(boolean hasBias)
- Parameters:
hasBias
- If true (default is false) the layer will have a bias
-
setInputSize
public LocallyConnected2D.Builder setInputSize(int... inputSize)
Set input filter size (h,w) for this locally connected 2D layer- Parameters:
inputSize
- pair of height and width of the input filters to this layer- Returns:
- Builder
-
build
public LocallyConnected2D build()
- Specified by:
build
in classLayer.Builder<LocallyConnected2D.Builder>
-
-