public abstract static class FeedForwardLayer.Builder<T extends FeedForwardLayer.Builder<T>> extends BaseLayer.Builder<T>
Modifier and Type | Field and Description |
---|---|
protected long |
nIn
Number of inputs for the layer (usually the size of the last layer).
|
protected long |
nOut
Number of inputs for the layer (usually the size of the last layer).
|
activationFn, biasInit, biasUpdater, gainInit, gradientNormalization, gradientNormalizationThreshold, iupdater, regularization, regularizationBias, weightInitFn, weightNoise
allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints
Constructor and Description |
---|
Builder() |
Modifier and Type | Method and Description |
---|---|
T |
nIn(int nIn)
Number of inputs for the layer (usually the size of the last layer).
|
T |
nIn(long nIn)
Number of inputs for the layer (usually the size of the last layer).
|
T |
nOut(int nOut)
Number of outputs - used to set the layer size (number of units/nodes for the current layer).
|
T |
nOut(long nOut)
Number of outputs - used to set the layer size (number of units/nodes for the current layer).
|
T |
units(int units)
Set the number of units / layer size for this layer.
This is equivalent to nOut(int) |
activation, activation, biasInit, biasUpdater, dist, gainInit, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, regularization, regularizationBias, updater, updater, weightDecay, weightDecay, weightDecayBias, weightDecayBias, weightInit, weightInit, weightInit, weightNoise
build, constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name
protected long nIn
protected long nOut
public T nIn(int nIn)
nIn
- Number of inputs for the layerpublic T nIn(long nIn)
nIn
- Number of inputs for the layerpublic T nOut(int nOut)
units(int)
nOut
- Number of outputs / layer sizepublic T nOut(long nOut)
units(int)
nOut
- Number of outputs / layer sizeCopyright © 2022. All rights reserved.