Class DepthwiseConvolution2D
- 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.ConvolutionLayer
-
- org.deeplearning4j.nn.conf.layers.DepthwiseConvolution2D
-
- All Implemented Interfaces:
Serializable
,Cloneable
,TrainingConfig
public class DepthwiseConvolution2D extends ConvolutionLayer
- See Also:
- Serialized Form
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
DepthwiseConvolution2D.Builder
-
Nested classes/interfaces inherited from class org.deeplearning4j.nn.conf.layers.ConvolutionLayer
ConvolutionLayer.AlgoMode, ConvolutionLayer.BaseConvBuilder<T extends ConvolutionLayer.BaseConvBuilder<T>>, ConvolutionLayer.BwdDataAlgo, ConvolutionLayer.BwdFilterAlgo, ConvolutionLayer.FwdAlgo
-
-
Field Summary
Fields Modifier and Type Field Description protected int
depthMultiplier
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.ConvolutionLayer
cnn2dDataFormat, convolutionMode, cudnnAlgoMode, cudnnAllowFallback, cudnnBwdDataAlgo, cudnnBwdFilterAlgo, cudnnFwdAlgo, dilation, hasBias, kernelSize, padding, stride
-
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 protected
DepthwiseConvolution2D(DepthwiseConvolution2D.Builder builder)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description DepthwiseConvolution2D
clone()
InputType
getOutputType(int layerIndex, InputType inputType)
For a given type of input to this layer, what is the type of the output?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.ConvolutionLayer
getMemoryReport, getPreProcessorForInputType, hasBias
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer
isPretrainParam
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer
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
-
-
-
-
Constructor Detail
-
DepthwiseConvolution2D
protected DepthwiseConvolution2D(DepthwiseConvolution2D.Builder builder)
-
-
Method Detail
-
clone
public DepthwiseConvolution2D clone()
- Overrides:
clone
in classConvolutionLayer
-
instantiate
public Layer instantiate(NeuralNetConfiguration conf, Collection<TrainingListener> trainingListeners, int layerIndex, INDArray layerParamsView, boolean initializeParams, DataType networkDataType)
- Overrides:
instantiate
in classConvolutionLayer
-
initializer
public ParamInitializer initializer()
- Overrides:
initializer
in classConvolutionLayer
- 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 classConvolutionLayer
- Parameters:
layerIndex
- Index of the layerinputType
- Type of input for the layer- Returns:
- Type of output from 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 classConvolutionLayer
- 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.
-
-