LSTMBlockCellGrad

public final class LSTMBlockCellGrad

Computes the LSTM cell backward propagation for 1 timestep.

This implementation is to be used in conjunction of LSTMBlockCell.

Constants

String

OP_NAME

The name of this op, as known by TensorFlow core engine

Public Methods

static <T extends TNumber> LSTMBlockCellGrad<T>	create(Scope scope, Operand<T> x, Operand<T> csPrev, Operand<T> hPrev, Operand<T> w, Operand<T> wci, Operand<T> wcf, Operand<T> wco, Operand<T> b, Operand<T> i, Operand<T> cs, Operand<T> f, Operand<T> o, Operand<T> ci, Operand<T> co, Operand<T> csGrad, Operand<T> hGrad, Boolean usePeephole) Factory method to create a class wrapping a new LSTMBlockCellGrad operation.
Output<T>	csPrevGrad() The gradient of cs to be back-propped.
Output<T>	dicfo() The derivative wrt to [i, cs, f, o].
Output<T>	wcfGrad() The gradient for wcf to be back-propped.
Output<T>	wciGrad() The gradient for wci to be back-propped.
Output<T>	wcoGrad() The gradient for wco to be back-propped.

Inherited Methods

From class org.tensorflow.op.RawOp

final boolean	equals(Object obj)
final int	hashCode()
Operation	op() Return this unit of computation as a single Operation.
final String	toString()

From class java.lang.Object

boolean	equals(Object arg0)
final Class<?>	getClass()
int	hashCode()
final void	notify()
final void	notifyAll()
String	toString()
final void	wait(long arg0, int arg1)
final void	wait(long arg0)
final void	wait()

From interface org.tensorflow.op.Op

abstract ExecutionEnvironment	env() Return the execution environment this op was created in.
abstract Operation	op() Return this unit of computation as a single Operation.

Constants

public static final String OP_NAME

The name of this op, as known by TensorFlow core engine

Constant Value: "LSTMBlockCellGrad"

Public Methods

public static LSTMBlockCellGrad<T> create (Scope scope, Operand<T> x, Operand<T> csPrev, Operand<T> hPrev, Operand<T> w, Operand<T> wci, Operand<T> wcf, Operand<T> wco, Operand<T> b, Operand<T> i, Operand<T> cs, Operand<T> f, Operand<T> o, Operand<T> ci, Operand<T> co, Operand<T> csGrad, Operand<T> hGrad, Boolean usePeephole)

Factory method to create a class wrapping a new LSTMBlockCellGrad operation.

Parameters

scope	current scope
x	The input to the LSTM cell, shape (batch_size, num_inputs).
csPrev	The previous cell state.
hPrev	The previous h state.
w	The weight matrix.
wci	The weight matrix for input gate peephole connection.
wcf	The weight matrix for forget gate peephole connection.
wco	The weight matrix for output gate peephole connection.
b	The bias vector.
i	The input gate.
cs	The cell state before the tanh.
f	The forget gate.
o	The output gate.
ci	The cell input.
co	The cell after the tanh.
csGrad	The current gradient of cs.
hGrad	The gradient of h vector.
usePeephole	Whether the cell uses peephole connections.

Returns

• a new instance of LSTMBlockCellGrad

public Output<T> csPrevGrad ()

The gradient of cs to be back-propped.

public Output<T> dicfo ()

The derivative wrt to [i, cs, f, o].

public Output<T> wcfGrad ()

The gradient for wcf to be back-propped.

public Output<T> wciGrad ()

The gradient for wci to be back-propped.

public Output<T> wcoGrad ()

The gradient for wco to be back-propped.