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 inherited from class org.deeplearning4j.nn.conf.layers.samediff.SameDiffLayer.Builder

weightInit

Fields inherited from class org.deeplearning4j.nn.conf.layers.samediff.AbstractSameDiffLayer.Builder

biasUpdater, l1, l1Bias, l2, l2Bias, 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	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)
LocallyConnected2D.Builder	setInputSize(int... inputSize) Set input filter size (h,w) for this locally connected 2D layer
LocallyConnected2D.Builder	stride(int... s)

Methods inherited from class org.deeplearning4j.nn.conf.layers.samediff.SameDiffLayer.Builder

weightInit

Methods inherited from class org.deeplearning4j.nn.conf.layers.samediff.AbstractSameDiffLayer.Builder

biasUpdater, l1, l1Bias, l2, l2Bias, updater

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()

Method Detail

nIn

public LocallyConnected2D.Builder nIn(int nIn)

nOut

public LocallyConnected2D.Builder nOut(int nOut)

activation

public LocallyConnected2D.Builder activation(Activation activation)

kernelSize

public LocallyConnected2D.Builder kernelSize(int... k)

stride

public LocallyConnected2D.Builder stride(int... s)

padding

public LocallyConnected2D.Builder padding(int... p)

convolutionMode

public LocallyConnected2D.Builder convolutionMode(ConvolutionMode cm)

dilation

public LocallyConnected2D.Builder dilation(int... d)

hasBias

public LocallyConnected2D.Builder hasBias(boolean hasBias)

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>