public class ComputationGraphUpdater
extends java.lang.Object
implements java.io.Serializable, java.lang.Cloneable
Updater
.Constructor and Description |
---|
ComputationGraphUpdater(ComputationGraph graph) |
ComputationGraphUpdater(ComputationGraph graph,
org.nd4j.linalg.api.ndarray.INDArray updaterState) |
Modifier and Type | Method and Description |
---|---|
ComputationGraphUpdater |
clone() |
boolean |
equals(java.lang.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
|
public ComputationGraphUpdater(ComputationGraph graph)
public ComputationGraphUpdater(ComputationGraph graph, org.nd4j.linalg.api.ndarray.INDArray updaterState)
public ComputationGraphUpdater clone()
clone
in class java.lang.Object
public void update(ComputationGraph graph, Gradient gradient, int iteration, int batchSize)
public void setStateViewArray(org.nd4j.linalg.api.ndarray.INDArray viewArray)
public org.nd4j.linalg.api.ndarray.INDArray getStateViewArray()
public boolean equals(java.lang.Object other)
equals
in class java.lang.Object
public int hashCode()
hashCode
in class java.lang.Object