org.deeplearning4j.nn.conf.layers

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 and 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 and Description
Builder()

Method Summary

Modifier and Type	Method and 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 layer
void	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

Methods inherited from class java.lang.Object

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

Field Detail

format

protected CNN2DFormat format

Constructor Detail

Builder

public Builder()

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 if ConvolutionMode.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 if ConvolutionMode.Same is set. Must be 2 values (height/width)

convolutionMode

public LocallyConnected2D.Builder convolutionMode(ConvolutionMode cm)

Parameters:

cm - Convolution mode for the layer. See ConvolutionMode 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). See CNN2DFormat 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 class Layer.Builder<LocallyConnected2D.Builder>