org.deeplearning4j.nn.conf.layers

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

public class DepthwiseConvolution2D
extends ConvolutionLayer

2D depth-wise convolution layer configuration.

Performs a channels-wise convolution, which operates on each of the input maps separately. A channel multiplier is used to specify the number of outputs per input map. This convolution is carried out with the specified kernel sizes, stride and padding values.

Author:

Max Pumperla

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and 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 inherited from class org.deeplearning4j.nn.conf.layers.ConvolutionLayer

convolutionMode, cudnnAlgoMode, cudnnAllowFallback, cudnnBwdDataAlgo, cudnnBwdFilterAlgo, cudnnFwdAlgo, dilation, hasBias, kernelSize, padding, stride

Fields inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer

nIn, nOut

Fields inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer

activationFn, biasInit, biasUpdater, dist, gradientNormalization, gradientNormalizationThreshold, iUpdater, l1, l1Bias, l2, l2Bias, weightInit, weightNoise

Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer

constraints, iDropout, layerName

Constructor Summary

Constructors

Modifier	Constructor and Description
protected	DepthwiseConvolution2D(DepthwiseConvolution2D.Builder builder)

Method Summary

Modifier and Type	Method and 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, org.nd4j.linalg.api.ndarray.INDArray layerParamsView, boolean initializeParams)

Methods inherited from class org.deeplearning4j.nn.conf.layers.ConvolutionLayer

getL1ByParam, getL2ByParam, getMemoryReport, getPreProcessorForInputType, hasBias, setNIn

Methods inherited from class org.deeplearning4j.nn.conf.layers.FeedForwardLayer

isPretrainParam

Methods inherited from class org.deeplearning4j.nn.conf.layers.BaseLayer

getUpdaterByParam, resetLayerDefaultConfig

Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer

initializeConstraints

Methods inherited from class java.lang.Object

equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

DepthwiseConvolution2D

protected DepthwiseConvolution2D(DepthwiseConvolution2D.Builder builder)

Method Detail

clone

public DepthwiseConvolution2D clone()

Overrides:

clone in class ConvolutionLayer

instantiate

public Layer instantiate(NeuralNetConfiguration conf,
                         Collection<TrainingListener> trainingListeners,
                         int layerIndex,
                         org.nd4j.linalg.api.ndarray.INDArray layerParamsView,
                         boolean initializeParams)

Overrides:

instantiate in class ConvolutionLayer

initializer

public ParamInitializer initializer()

Overrides:

initializer in class ConvolutionLayer

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 class ConvolutionLayer

Parameters:

layerIndex - Index of the layer

inputType - Type of input for the layer

Returns:

Type of output from the layer