org.deeplearning4j.nn.conf.layers

Class BaseOutputLayer

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.BaseOutputLayer

All Implemented Interfaces:

Serializable, Cloneable

Direct Known Subclasses:

CenterLossOutputLayer, OutputLayer, RnnOutputLayer

public abstract class BaseOutputLayer
extends FeedForwardLayer

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	BaseOutputLayer.Builder<T extends BaseOutputLayer.Builder<T>>

Field Summary

Fields

Modifier and Type	Field and Description
protected org.nd4j.linalg.lossfunctions.ILossFunction	lossFn

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

nIn, nOut

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

activationFn, adamMeanDecay, adamVarDecay, biasInit, biasLearningRate, dist, epsilon, gradientNormalization, gradientNormalizationThreshold, iUpdater, l1, l1Bias, l2, l2Bias, learningRate, learningRateSchedule, momentum, momentumSchedule, rho, rmsDecay, updater, weightInit

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

dropOut, layerName

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction	getLossFunction() Deprecated. As of 0.6.0. Use #getLossFn() instead
LayerMemoryReport	getMemoryReport(InputType inputType) This is a report of the estimated memory consumption for the given layer

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

getL1ByParam, getL2ByParam, getLearningRateByParam, getOutputType, getPreProcessorForInputType, isPretrainParam, setNIn

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

clone, getIUpdaterByParam, getUpdaterByParam, resetLayerDefaultConfig

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

initializer, instantiate

Methods inherited from class java.lang.Object

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

Field Detail

lossFn

protected org.nd4j.linalg.lossfunctions.ILossFunction lossFn

Constructor Detail

BaseOutputLayer

protected BaseOutputLayer(BaseOutputLayer.Builder builder)

Method Detail

getLossFunction

@Deprecated
public org.nd4j.linalg.lossfunctions.LossFunctions.LossFunction getLossFunction()

Deprecated. As of 0.6.0. Use #getLossFn() instead

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

Parameters:

inputType - Input type to the layer. Memory consumption is often a function of the input type

Returns:

Memory report for the layer