Class CnnLossLayer
- 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.CnnLossLayer
-
- All Implemented Interfaces:
Serializable,Cloneable,TrainingConfig
public class CnnLossLayer extends FeedForwardLayer
- See Also:
- Serialized Form
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static classCnnLossLayer.Builder
-
Field Summary
Fields Modifier and Type Field Description protected CNN2DFormatformatprotected 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
-
-
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 layerInputTypegetOutputType(int layerIndex, InputType inputType)For a given type of input to this layer, what is the type of the output?InputPreProcessorgetPreProcessorForInputType(InputType inputType)For the given type of input to this layer, what preprocessor (if any) is required?
Returns null if no preprocessor is required, otherwise returns an appropriateInputPreProcessorfor this layer, such as aCnnToFeedForwardPreProcessorParamInitializerinitializer()Layerinstantiate(NeuralNetConfiguration conf, Collection<TrainingListener> trainingListeners, int layerIndex, INDArray layerParamsView, boolean initializeParams, DataType networkDataType)voidsetNIn(InputType inputType, boolean override)Set the nIn value (number of inputs, or input channels for CNNs) based on the given input type-
Methods inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer
isPretrainParam
-
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, 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
-
format
protected CNN2DFormat format
-
-
Method Detail
-
instantiate
public Layer instantiate(NeuralNetConfiguration conf, Collection<TrainingListener> trainingListeners, int layerIndex, INDArray layerParamsView, boolean initializeParams, DataType networkDataType)
- Specified by:
instantiatein classLayer
-
initializer
public ParamInitializer initializer()
- Specified by:
initializerin classLayer- Returns:
- The parameter initializer for this model
-
getOutputType
public InputType getOutputType(int layerIndex, InputType inputType)
Description copied from class:LayerFor a given type of input to this layer, what is the type of the output?- Overrides:
getOutputTypein classFeedForwardLayer- Parameters:
layerIndex- Index of the layerinputType- Type of input for the layer- Returns:
- Type of output from the layer
-
getPreProcessorForInputType
public InputPreProcessor getPreProcessorForInputType(InputType inputType)
Description copied from class:LayerFor the given type of input to this layer, what preprocessor (if any) is required?
Returns null if no preprocessor is required, otherwise returns an appropriateInputPreProcessorfor this layer, such as aCnnToFeedForwardPreProcessor- Overrides:
getPreProcessorForInputTypein classFeedForwardLayer- Parameters:
inputType- InputType to this layer- Returns:
- Null if no preprocessor is required, otherwise the type of preprocessor necessary for this layer/input combination
-
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
-
setNIn
public void setNIn(InputType inputType, boolean override)
Description copied from class:LayerSet the nIn value (number of inputs, or input channels for CNNs) based on the given input type- Overrides:
setNInin classFeedForwardLayer- Parameters:
inputType- Input type for this layeroverride- If false: only set the nIn value if it's not already set. If true: set it regardless of whether it's already set or not.
-
-