T
- data type for csPrevGrad()
outputpublic final class LSTMBlockCellGrad<T extends Number> extends PrimitiveOp
This implementation is to be used in conjunction of LSTMBlockCell.
operation
Modifier and Type | Method and Description |
---|---|
static <T extends Number> |
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.
|
equals, hashCode, op, toString
public static <T extends Number> 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)
scope
- current scopex
- 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.Copyright © 2022. All rights reserved.