org.deeplearning4j.nn.gradient

Class DefaultGradient

java.lang.Object

org.deeplearning4j.nn.gradient.DefaultGradient

All Implemented Interfaces:

Serializable, Gradient

public class DefaultGradient
extends Object
implements Gradient

Default gradient implementation. Basically lookup table for ndarrays

Author:

Adam Gibson

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
DefaultGradient()

Method Summary

Modifier and Type	Method and Description
void	clear() Clear residual parameters (useful for returning a gradient and then clearing old objects)
org.nd4j.linalg.api.ndarray.INDArray	getGradientFor(String variable) The gradient for the given variable
org.nd4j.linalg.api.ndarray.INDArray	gradient() The full gradient as one flat vector
org.nd4j.linalg.api.ndarray.INDArray	gradient(List<String> order) The full gradient as one flat vector
Map<String,org.nd4j.linalg.api.ndarray.INDArray>	gradientForVariable() Gradient look up table
String	toString()

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Detail

DefaultGradient

public DefaultGradient()

Method Detail

gradientForVariable

public Map<String,org.nd4j.linalg.api.ndarray.INDArray> gradientForVariable()

Description copied from interface: Gradient

Gradient look up table

Specified by:

gradientForVariable in interface Gradient

Returns:

the gradient look up table

gradient

public org.nd4j.linalg.api.ndarray.INDArray gradient(List<String> order)

Description copied from interface: Gradient

The full gradient as one flat vector

Specified by:

gradient in interface Gradient

Returns:

gradient

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

Description copied from interface: Gradient

The full gradient as one flat vector

Specified by:

gradient in interface Gradient

Returns:

clear

public void clear()

Description copied from interface: Gradient

Clear residual parameters (useful for returning a gradient and then clearing old objects)

Specified by:

clear in interface Gradient

getGradientFor

public org.nd4j.linalg.api.ndarray.INDArray getGradientFor(String variable)

Description copied from interface: Gradient

The gradient for the given variable

Specified by:

getGradientFor in interface Gradient

Parameters:

variable - the variable to get the gradient for

Returns:

the gradient for the given variable or null

toString

public String toString()

Overrides:

toString in class Object