org.deeplearning4j.nn.updater

Class LayerUpdater

java.lang.Object

org.deeplearning4j.nn.updater.BaseMultiLayerUpdater<Layer>

org.deeplearning4j.nn.updater.LayerUpdater

All Implemented Interfaces:

Serializable, Updater

public class LayerUpdater
extends BaseMultiLayerUpdater<Layer>

Updater for a single layer, excluding MultiLayerNetwork (which also implements the Layer interface)

Author:

Alex Black

See Also:

Serialized Form

Field Summary

Fields inherited from class org.deeplearning4j.nn.updater.BaseMultiLayerUpdater

gradientsForMinibatchDivision, initializedMinibatchDivision, layersByName, network, updaterBlocks, updaterStateViewArray

Constructor Summary

Constructors

Constructor and Description
LayerUpdater(Layer layer)
LayerUpdater(Layer layer, INDArray updaterState)

Method Summary

Modifier and Type	Method and Description
protected INDArray	getFlattenedGradientsView()
protected Trainable[]	getOrderedLayers()
protected INDArray	getParams()
protected boolean	isMiniBatch()
protected boolean	isSingleLayerUpdater()

Methods inherited from class org.deeplearning4j.nn.updater.BaseMultiLayerUpdater

divideByMinibatch, equals, getMinibatchDivisionSubsets, getStateViewArray, getStateViewArrayCopy, hashCode, preApply, setStateViewArray, setStateViewArray, update, update

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

LayerUpdater

public LayerUpdater(Layer layer)

LayerUpdater

public LayerUpdater(Layer layer,
                    INDArray updaterState)

Method Detail

getOrderedLayers

protected Trainable[] getOrderedLayers()

Specified by:

getOrderedLayers in class BaseMultiLayerUpdater<Layer>

Returns:

Array of layers, in the correct order (i.e., same order as the parameter/gradient/updater flattening order - input to output for MultiLayerNetwork, or topological order for ComputationGraph)

getFlattenedGradientsView

protected INDArray getFlattenedGradientsView()

Specified by:

getFlattenedGradientsView in class BaseMultiLayerUpdater<Layer>

Returns:

The flattened gradient view array for the model

getParams

protected INDArray getParams()

Specified by:

getParams in class BaseMultiLayerUpdater<Layer>

Returns:

The flattened parameter array for the model

isMiniBatch

protected boolean isMiniBatch()

Specified by:

isMiniBatch in class BaseMultiLayerUpdater<Layer>

Returns:

True if the configuration for the model is set to minibatch (divide by minibatch size), false otherwise

isSingleLayerUpdater

protected boolean isSingleLayerUpdater()

Overrides:

isSingleLayerUpdater in class BaseMultiLayerUpdater<Layer>