Class BaseOutputLayer
- java.lang.Object
-
- org.deeplearning4j.nn.conf.layers.Layer
-
- org.deeplearning4j.nn.conf.layers.BaseLayer
-
- org.deeplearning4j.nn.conf.layers.FeedForwardLayer
-
- org.deeplearning4j.nn.conf.layers.BaseOutputLayer
-
- All Implemented Interfaces:
Serializable,Cloneable,TrainingConfig
- Direct Known Subclasses:
CenterLossOutputLayer,OCNNOutputLayer,OutputLayer,RnnOutputLayer
public abstract class BaseOutputLayer extends FeedForwardLayer
- See Also:
- Serialized Form
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static classBaseOutputLayer.Builder<T extends BaseOutputLayer.Builder<T>>
-
Field Summary
Fields Modifier and Type Field Description protected booleanhasBiasprotected ILossFunctionlossFn-
Fields inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer
nIn, nOut, timeDistributedFormat
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer
activationFn, biasInit, biasUpdater, gainInit, gradientNormalization, gradientNormalizationThreshold, iUpdater, regularization, regularizationBias, weightInitFn, weightNoise
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer
constraints, iDropout, layerName
-
-
Constructor Summary
Constructors Modifier Constructor Description protectedBaseOutputLayer(BaseOutputLayer.Builder builder)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description LayerMemoryReportgetMemoryReport(InputType inputType)This is a report of the estimated memory consumption for the given layerbooleanhasBias()-
Methods inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer
getOutputType, getPreProcessorForInputType, isPretrainParam, setNIn
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer
clone, getGradientNormalization, getRegularizationByParam, getUpdaterByParam, resetLayerDefaultConfig
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer
initializeConstraints, initializer, instantiate, setDataType
-
Methods inherited from class java.lang.Object
equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
-
Methods inherited from interface org.deeplearning4j.nn.api.TrainingConfig
getGradientNormalizationThreshold, getLayerName
-
-
-
-
Field Detail
-
lossFn
protected ILossFunction lossFn
-
hasBias
protected boolean hasBias
-
-
Constructor Detail
-
BaseOutputLayer
protected BaseOutputLayer(BaseOutputLayer.Builder builder)
-
-
Method Detail
-
hasBias
public boolean hasBias()
-
getMemoryReport
public LayerMemoryReport getMemoryReport(InputType inputType)
Description copied from class:LayerThis is a report of the estimated memory consumption for the given layer- Specified by:
getMemoryReportin classLayer- Parameters:
inputType- Input type to the layer. Memory consumption is often a function of the input type- Returns:
- Memory report for the layer
-
-