org.deeplearning4j.nn.updater.graph

Class ComputationGraphUpdater

java.lang.Object

org.deeplearning4j.nn.updater.graph.ComputationGraphUpdater

All Implemented Interfaces:

Serializable, Cloneable

public class ComputationGraphUpdater
extends Object
implements Serializable, Cloneable

Gradient updater for ComputationGraph.
Note: ComputationGraph does not implement the Layer interface (due to multiple in/out etc), hence ComputationGraphUpdater can't be defined as an Updater.

Author:

Alex Black

See Also:

Serialized Form

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
ComputationGraphUpdater	clone()
boolean	equals(Object other)
org.nd4j.linalg.api.ndarray.INDArray	getStateViewArray()
int	hashCode()
void	setStateViewArray(org.nd4j.linalg.api.ndarray.INDArray viewArray)
void	update(ComputationGraph graph, Gradient gradient, int iteration, int batchSize) Update the gradients for the given ComputationGraph

Methods inherited from class java.lang.Object

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

Constructor Detail

ComputationGraphUpdater

public ComputationGraphUpdater(ComputationGraph graph)

ComputationGraphUpdater

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

Method Detail

clone

public ComputationGraphUpdater clone()

Overrides:

clone in class Object

update

public void update(ComputationGraph graph,
                   Gradient gradient,
                   int iteration,
                   int batchSize)

Update the gradients for the given ComputationGraph

setStateViewArray

public void setStateViewArray(org.nd4j.linalg.api.ndarray.INDArray viewArray)

getStateViewArray

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

equals

public boolean equals(Object other)

Overrides:

equals in class Object

hashCode

public int hashCode()

Overrides:

hashCode in class Object