public static class Cnn3DLossLayer.Builder extends BaseOutputLayer.Builder<Cnn3DLossLayer.Builder>
Modifier and Type | Field and Description |
---|---|
protected Convolution3D.DataFormat |
dataFormat
Format of the input/output data.
|
lossFn
nIn, nOut
activationFn, biasInit, biasUpdater, gainInit, gradientNormalization, gradientNormalizationThreshold, iupdater, regularization, regularizationBias, weightInitFn, weightNoise
allParamConstraints, biasConstraints, iDropout, layerName, weightConstraints
Constructor and Description |
---|
Builder(Convolution3D.DataFormat format) |
Modifier and Type | Method and Description |
---|---|
Cnn3DLossLayer |
build() |
Cnn3DLossLayer.Builder |
nIn(int nIn)
Number of inputs for the layer (usually the size of the last layer).
|
Cnn3DLossLayer.Builder |
nOut(int nOut)
Number of outputs - used to set the layer size (number of units/nodes for the current layer).
|
void |
setNIn(long nIn) |
void |
setNOut(long nOut) |
hasBias, lossFunction, lossFunction
nIn, nOut, units
activation, activation, biasInit, biasUpdater, dist, gainInit, gradientNormalization, gradientNormalizationThreshold, l1, l1Bias, l2, l2Bias, regularization, regularizationBias, updater, updater, weightDecay, weightDecay, weightDecayBias, weightDecayBias, weightInit, weightInit, weightInit, weightNoise
constrainAllParameters, constrainBias, constrainWeights, dropOut, dropOut, name
protected Convolution3D.DataFormat dataFormat
Convolution3D.DataFormat
for detailspublic Builder(@NonNull Convolution3D.DataFormat format)
format
- Format of the input/output data. See Convolution3D.DataFormat
for detailspublic Cnn3DLossLayer.Builder nIn(int nIn)
FeedForwardLayer.Builder
nIn
in class FeedForwardLayer.Builder<Cnn3DLossLayer.Builder>
nIn
- Number of inputs for the layerpublic Cnn3DLossLayer.Builder nOut(int nOut)
FeedForwardLayer.Builder
FeedForwardLayer.Builder.units(int)
nOut
in class FeedForwardLayer.Builder<Cnn3DLossLayer.Builder>
nOut
- Number of outputs / layer sizepublic void setNIn(long nIn)
public void setNOut(long nOut)
public Cnn3DLossLayer build()
build
in class Layer.Builder<Cnn3DLossLayer.Builder>
Copyright © 2022. All rights reserved.