org.deeplearning4j.nn.updater

Class MultiLayerUpdater

java.lang.Object

org.deeplearning4j.nn.updater.MultiLayerUpdater

All Implemented Interfaces:

java.io.Serializable, java.lang.Cloneable, Updater

public class MultiLayerUpdater
extends java.lang.Object
implements Updater

MultiLayerUpdater: Gradient updater for MultiLayerNetworks. Expects backprop gradients for all layers to be in single Gradient object, keyed by "0_b", "1_w" etc., as per MultiLayerNetwork.backward()

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
MultiLayerUpdater(MultiLayerNetwork network)
MultiLayerUpdater(MultiLayerNetwork network, org.nd4j.linalg.api.ndarray.INDArray updaterState)

Method Summary

Modifier and Type	Method and Description
Updater	clone()
boolean	equals(java.lang.Object other)
org.nd4j.linalg.api.ndarray.INDArray	getStateViewArray()
void	setStateViewArray(Layer layer, org.nd4j.linalg.api.ndarray.INDArray viewArray, boolean initialize) Set the internal (historical) state view array for this updater
int	stateSizeForLayer(Layer layer) Calculate and return the state size for this updater (for the given layer).
void	update(Layer layer, Gradient gradient, int iteration, int batchSize) Updater: updates the model

Methods inherited from class java.lang.Object

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

Constructor Detail

MultiLayerUpdater

public MultiLayerUpdater(MultiLayerNetwork network)

MultiLayerUpdater

public MultiLayerUpdater(MultiLayerNetwork network,
                         org.nd4j.linalg.api.ndarray.INDArray updaterState)

Method Detail

setStateViewArray

public void setStateViewArray(Layer layer,
                              org.nd4j.linalg.api.ndarray.INDArray viewArray,
                              boolean initialize)

Description copied from interface: Updater

Set the internal (historical) state view array for this updater

Specified by:

setStateViewArray in interface Updater

Parameters:

layer - Layer that this updater belongs to

viewArray - View array

initialize - Whether to initialize the array or not

getStateViewArray

public org.nd4j.linalg.api.ndarray.INDArray getStateViewArray()

Specified by:

getStateViewArray in interface Updater

Returns:

the view array for this updater

stateSizeForLayer

public int stateSizeForLayer(Layer layer)

Description copied from interface: Updater

Calculate and return the state size for this updater (for the given layer). How many parameters/values does this updater have?

Specified by:

stateSizeForLayer in interface Updater

Parameters:

layer - Layer that this updater belongs to

Returns:

number of parameters/values in the updater state

update

public void update(Layer layer,
                   Gradient gradient,
                   int iteration,
                   int batchSize)

Description copied from interface: Updater

Updater: updates the model

Specified by:

update in interface Updater

clone

public Updater clone()

Specified by:

clone in interface Updater

Overrides:

clone in class java.lang.Object

equals

public boolean equals(java.lang.Object other)

Overrides:

equals in class java.lang.Object