org.deeplearning4j.nn.gradient

Interface Gradient

All Superinterfaces:

java.io.Serializable

All Known Implementing Classes:

DefaultGradient

public interface Gradient
extends java.io.Serializable

Generic gradient

Method Summary

Modifier and Type	Method and Description
void	clear() Clear residual parameters (useful for returning a gradient and then clearing old objects)
java.lang.Character	flatteningOrderForVariable(java.lang.String variable) Return the gradient flattening order for the specified variable, or null if it is not explicitly set
org.nd4j.linalg.api.ndarray.INDArray	getGradientFor(java.lang.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(java.util.List<java.lang.String> order) The full gradient as one flat vector
java.util.Map<java.lang.String,org.nd4j.linalg.api.ndarray.INDArray>	gradientForVariable() Gradient look up table
org.nd4j.linalg.api.ndarray.INDArray	setGradientFor(java.lang.String variable, org.nd4j.linalg.api.ndarray.INDArray gradient) Update gradient for the given variable
org.nd4j.linalg.api.ndarray.INDArray	setGradientFor(java.lang.String variable, org.nd4j.linalg.api.ndarray.INDArray gradient, java.lang.Character flatteningOrder) Update gradient for the given variable; also (optionally) specify the order in which the array should be flattened to a row vector

Method Detail

gradientForVariable

java.util.Map<java.lang.String,org.nd4j.linalg.api.ndarray.INDArray> gradientForVariable()

Gradient look up table

Returns:

the gradient look up table

gradient

org.nd4j.linalg.api.ndarray.INDArray gradient(java.util.List<java.lang.String> order)

The full gradient as one flat vector

Returns:

gradient

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

The full gradient as one flat vector

Returns:

clear

void clear()

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

getGradientFor

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

The gradient for the given variable

Parameters:

variable - the variable to get the gradient for

Returns:

the gradient for the given variable or null

setGradientFor

org.nd4j.linalg.api.ndarray.INDArray setGradientFor(java.lang.String variable,
                                                    org.nd4j.linalg.api.ndarray.INDArray gradient)

Update gradient for the given variable

Parameters:

variable - the variable to get the gradient for

gradient - the gradient values

Returns:

the gradient for the given variable or null

setGradientFor

org.nd4j.linalg.api.ndarray.INDArray setGradientFor(java.lang.String variable,
                                                    org.nd4j.linalg.api.ndarray.INDArray gradient,
                                                    java.lang.Character flatteningOrder)

Update gradient for the given variable; also (optionally) specify the order in which the array should be flattened to a row vector

Parameters:

variable - the variable to get the gradient for

gradient - the gradient values

flatteningOrder - the order in which gradients should be flattened (null ok - default)

Returns:

the gradient for the given variable or null

flatteningOrderForVariable

java.lang.Character flatteningOrderForVariable(java.lang.String variable)

Return the gradient flattening order for the specified variable, or null if it is not explicitly set

Parameters:

variable - Variable to return the gradient flattening order for

Returns:

Order in which the specified variable's gradient should be flattened