Class Cnn3DLossLayer
- 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.Cnn3DLossLayer
-
- All Implemented Interfaces:
Serializable
,Cloneable
,TrainingConfig
public class Cnn3DLossLayer extends FeedForwardLayer
- See Also:
- Serialized Form
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
Cnn3DLossLayer.Builder
-
Field Summary
Fields Modifier and Type Field Description protected Convolution3D.DataFormat
dataFormat
protected ILossFunction
lossFn
-
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 LayerMemoryReport
getMemoryReport(InputType inputType)
This is a report of the estimated memory consumption for the given layerInputType
getOutputType(int layerIndex, InputType inputType)
For a given type of input to this layer, what is the type of the output?InputPreProcessor
getPreProcessorForInputType(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 appropriateInputPreProcessor
for this layer, such as aCnnToFeedForwardPreProcessor
ParamInitializer
initializer()
Layer
instantiate(NeuralNetConfiguration conf, Collection<TrainingListener> trainingListeners, int layerIndex, INDArray layerParamsView, boolean initializeParams, DataType networkDataType)
void
setNIn(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
-
dataFormat
protected Convolution3D.DataFormat dataFormat
-
-
Method Detail
-
instantiate
public Layer instantiate(NeuralNetConfiguration conf, Collection<TrainingListener> trainingListeners, int layerIndex, INDArray layerParamsView, boolean initializeParams, DataType networkDataType)
- Specified by:
instantiate
in classLayer
-
initializer
public ParamInitializer initializer()
- Specified by:
initializer
in classLayer
- Returns:
- The parameter initializer for this model
-
getOutputType
public InputType getOutputType(int layerIndex, InputType inputType)
Description copied from class:Layer
For a given type of input to this layer, what is the type of the output?- Overrides:
getOutputType
in 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:Layer
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 appropriateInputPreProcessor
for this layer, such as aCnnToFeedForwardPreProcessor
- Overrides:
getPreProcessorForInputType
in 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:Layer
This is a report of the estimated memory consumption for the given layer- Specified by:
getMemoryReport
in 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:Layer
Set the nIn value (number of inputs, or input channels for CNNs) based on the given input type- Overrides:
setNIn
in 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.
-
-