Class CapsuleStrengthLayer
- java.lang.Object
-
- org.deeplearning4j.nn.conf.layers.Layer
-
- org.deeplearning4j.nn.conf.layers.samediff.AbstractSameDiffLayer
-
- org.deeplearning4j.nn.conf.layers.samediff.SameDiffLayer
-
- org.deeplearning4j.nn.conf.layers.samediff.SameDiffLambdaLayer
-
- org.deeplearning4j.nn.conf.layers.CapsuleStrengthLayer
-
- All Implemented Interfaces:
Serializable
,Cloneable
,TrainingConfig
public class CapsuleStrengthLayer extends SameDiffLambdaLayer
- See Also:
- Serialized Form
-
-
Nested Class Summary
Nested Classes Modifier and Type Class Description static class
CapsuleStrengthLayer.Builder
-
Field Summary
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.samediff.SameDiffLayer
paramWeightInit, weightInit
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.samediff.AbstractSameDiffLayer
biasUpdater, gradientNormalization, gradientNormalizationThreshold, regularization, regularizationBias, updater
-
Fields inherited from class org.deeplearning4j.nn.conf.layers.Layer
constraints, iDropout, layerName
-
-
Constructor Summary
Constructors Constructor Description CapsuleStrengthLayer(CapsuleStrengthLayer.Builder builder)
-
Method Summary
All Methods Instance Methods Concrete Methods Modifier and Type Method Description SDVariable
defineLayer(SameDiff SD, SDVariable layerInput)
The defineLayer method is used to define the forward pass for the layerInputType
getOutputType(int layerIndex, InputType inputType)
For a given type of input to this layer, what is the type of the output?-
Methods inherited from class org.deeplearning4j.nn.conf.layers.samediff.SameDiffLambdaLayer
defineLayer, defineParameters, initializeParameters
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.samediff.SameDiffLayer
feedForwardMaskArray, instantiate, validateInput
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.samediff.AbstractSameDiffLayer
applyGlobalConfig, applyGlobalConfigToLayer, getLayerParams, getMemoryReport, getPreProcessorForInputType, getRegularizationByParam, getUpdaterByParam, initializer, initWeights, isPretrainParam, onesMaskForInput, paramReshapeOrder, setNIn
-
Methods inherited from class org.deeplearning4j.nn.conf.layers.Layer
clone, initializeConstraints, resetLayerDefaultConfig, 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
getGradientNormalization, getGradientNormalizationThreshold, getLayerName
-
-
-
-
Constructor Detail
-
CapsuleStrengthLayer
public CapsuleStrengthLayer(CapsuleStrengthLayer.Builder builder)
-
-
Method Detail
-
defineLayer
public SDVariable defineLayer(SameDiff SD, SDVariable layerInput)
Description copied from class:SameDiffLambdaLayer
The defineLayer method is used to define the forward pass for the layer- Specified by:
defineLayer
in classSameDiffLambdaLayer
- Parameters:
SD
- SameDiff instance to use to define the vertexlayerInput
- Layer input variable- Returns:
- The output variable (corresponding to the output activations for the layer)
-
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 classSameDiffLambdaLayer
- Parameters:
layerIndex
- Index of the layerinputType
- Type of input for the layer- Returns:
- Type of output from the layer
-
-