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

layersByName, network, updaterBlocks, updaterStateViewArray

Constructor Summary

Constructors

Constructor and Description
LayerUpdater(Layer layer)
LayerUpdater(Layer layer, org.nd4j.linalg.api.ndarray.INDArray updaterState)

Method Summary

Modifier and Type	Method and Description
protected org.nd4j.linalg.api.ndarray.INDArray	getFlattenedGradientsView()
protected Layer[]	getOrderedLayers()
protected org.nd4j.linalg.api.ndarray.INDArray	getParams()
protected boolean	isMiniBatch()
protected boolean	isSingleLayerUpdater()

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

equals, getStateViewArray, 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,
                    org.nd4j.linalg.api.ndarray.INDArray updaterState)

Method Detail

getOrderedLayers

protected Layer[] 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 org.nd4j.linalg.api.ndarray.INDArray getFlattenedGradientsView()

Specified by:

getFlattenedGradientsView in class BaseMultiLayerUpdater<Layer>

Returns:

The flattened gradient view array for the model

getParams

protected org.nd4j.linalg.api.ndarray.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>