org.deeplearning4j.nn.updater.graph

Class ComputationGraphUpdater

java.lang.Object

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

org.deeplearning4j.nn.updater.graph.ComputationGraphUpdater

All Implemented Interfaces:

Serializable, Updater

public class ComputationGraphUpdater
extends BaseMultiLayerUpdater<ComputationGraph>

Gradient updater for ComputationGraph. Most of the functionality is shared with MultiLayerUpdater via BaseMultiLayerUpdater

Author:

Alex Black

See Also:

Serialized Form

Field Summary

Fields

Modifier and Type	Field and Description
protected Layer[]	orderedLayers

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

layersByName, network, updaterBlocks, updaterStateViewArray

Constructor Summary

Constructors

Constructor and Description
ComputationGraphUpdater(ComputationGraph graph)
ComputationGraphUpdater(ComputationGraph graph, 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()

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

equals, getStateViewArray, hashCode, isSingleLayerUpdater, preApply, setStateViewArray, setStateViewArray, update, update

Methods inherited from class java.lang.Object

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

Field Detail

orderedLayers

protected Layer[] orderedLayers

Constructor Detail

ComputationGraphUpdater

public ComputationGraphUpdater(ComputationGraph graph)

ComputationGraphUpdater

public ComputationGraphUpdater(ComputationGraph graph,
                               org.nd4j.linalg.api.ndarray.INDArray updaterState)

Method Detail

getOrderedLayers

protected Layer[] getOrderedLayers()

Specified by:

getOrderedLayers in class BaseMultiLayerUpdater<ComputationGraph>

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<ComputationGraph>

Returns:

The flattened gradient view array for the model

getParams

protected org.nd4j.linalg.api.ndarray.INDArray getParams()

Specified by:

getParams in class BaseMultiLayerUpdater<ComputationGraph>

Returns:

The flattened parameter array for the model

isMiniBatch

protected boolean isMiniBatch()

Specified by:

isMiniBatch in class BaseMultiLayerUpdater<ComputationGraph>

Returns:

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