org.deeplearning4j.nn.layers.feedforward.dense

Class DenseLayer

java.lang.Object

org.deeplearning4j.nn.layers.BaseLayer<DenseLayer>

org.deeplearning4j.nn.layers.feedforward.dense.DenseLayer

All Implemented Interfaces:

java.io.Serializable, java.lang.Cloneable, Layer, Model

public class DenseLayer
extends BaseLayer<DenseLayer>

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from interface org.deeplearning4j.nn.api.Layer

Layer.TrainingMode, Layer.Type

Field Summary

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

conf, dropoutApplied, dropoutMask, gradient, gradientsFlattened, gradientViews, index, input, iterationListeners, maskArray, maskState, optimizer, params, paramsFlattened, score, solver

Constructor Summary

Constructors

Constructor and Description
DenseLayer(NeuralNetConfiguration conf)
DenseLayer(NeuralNetConfiguration conf, org.nd4j.linalg.api.ndarray.INDArray input)

Method Summary

Modifier and Type	Method and Description
void	fit(org.nd4j.linalg.api.ndarray.INDArray input) Fit the model to the given data
boolean	isPretrainLayer() Returns true if the layer can be trained in an unsupervised/pretrain manner (VAE, RBMs etc)

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

accumulateScore, activate, activate, activate, activate, activate, activate, activationMean, applyDropOutIfNecessary, applyLearningRateScoreDecay, applyMask, backpropGradient, batchSize, calcGradient, calcL1, calcL2, clear, clone, computeGradientAndScore, conf, createGradient, derivativeActivation, error, feedForwardMaskArray, fit, getIndex, getInput, getInputMiniBatchSize, getListeners, getMaskArray, getOptimizer, getParam, gradient, gradientAndScore, init, initParams, input, iterate, layerConf, layerNameAndIndex, merge, numParams, numParams, params, paramTable, paramTable, preOutput, preOutput, preOutput, preOutput, score, setBackpropGradientsViewArray, setConf, setIndex, setInput, setInputMiniBatchSize, setListeners, setListeners, setMaskArray, setParam, setParams, setParams, setParamsViewArray, setParamTable, setScoreWithZ, toString, transpose, type, update, update, validateInput

Methods inherited from class java.lang.Object

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

Constructor Detail

DenseLayer

public DenseLayer(NeuralNetConfiguration conf)

DenseLayer

public DenseLayer(NeuralNetConfiguration conf,
                  org.nd4j.linalg.api.ndarray.INDArray input)

Method Detail

fit

public void fit(org.nd4j.linalg.api.ndarray.INDArray input)

Description copied from interface: Model

Fit the model to the given data

Specified by:

fit in interface Model

Overrides:

fit in class BaseLayer<DenseLayer>

Parameters:

input - the data to fit the model to

isPretrainLayer

public boolean isPretrainLayer()

Description copied from interface: Layer

Returns true if the layer can be trained in an unsupervised/pretrain manner (VAE, RBMs etc)

Returns:

true if the layer can be pretrained (using fit(INDArray), false otherwise