org.deeplearning4j.nn.api

Interface Updater

All Superinterfaces:

java.lang.Cloneable, java.io.Serializable

All Known Implementing Classes:

LayerUpdater, MultiLayerUpdater

public interface Updater
extends java.io.Serializable, java.lang.Cloneable

Update the model

Method Summary

Modifier and Type	Method and Description
Updater	clone()
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 miniBatchSize) Updater: updates the model

Method Detail

setStateViewArray

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

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

Parameters:

layer - Layer that this updater belongs to

viewArray - View array

initialize - Whether to initialize the array or not

getStateViewArray

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

Returns:

the view array for this updater

stateSizeForLayer

int stateSizeForLayer(Layer layer)

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

Parameters:

layer - Layer that this updater belongs to

Returns:

number of parameters/values in the updater state

update

void update(Layer layer,
            Gradient gradient,
            int iteration,
            int miniBatchSize)

Updater: updates the model

Parameters:

layer -

gradient -

iteration -

clone

Updater clone()