tf.raw_ops.LSTMBlockCellGrad
Вычисляет обратное распространение LSTM-ячейки на 1 шаг.
tf.raw_ops.LSTMBlockCellGrad(
x,
cs_prev,
h_prev,
w,
wci,
wcf,
wco,
b,
i,
cs,
f,
o,
ci,
co,
cs_grad,
h_grad,
use_peephole,
name=None
)
Это реализация должна использоваться совместно с LSTMBlockCell.
| Аргументы | |
|---|---|
x | A Tensor. Должен быть одного из следующих типов: half, float32. Вход в LSTM-ячейку, форма (размерность_пакета, число_входов). |
cs_prev | A Tensor. Должен иметь тот же тип, что и x. Предыдущее состояние ячейки. |
h_prev | A Tensor. Должен иметь тот же тип, что и x. Предыдущее состояние h. |
w | A Tensor. Должен иметь тот же тип, что и x. Матрица весов. |
wci | A Tensor. Должен иметь тот же тип, что и x. Матрица весов для связи с просматривающейся ячейкой входной двери. |
wcf | A Tensor. Должен иметь тот же тип, что и x. Матрица весов для связи с просматривающейся ячейкой входной двери. |
wco | A Tensor. Должен иметь тот же тип, что и x. Матрица весов для связи с просматривающейся ячейкой выходной двери. |
b | A Tensor. Должен иметь тот же тип, что и x. Вектор смещения. |
i | A Tensor. Должен иметь тот же тип, что и x. Входная дверь. |
cs | A Tensor. Должен иметь тот же тип, что и x. Состояние ячейки до tanh. |
f | A Tensor. Должен иметь тот же тип, что и x. Забывающая дверь. |
o | A Tensor. Должен иметь тот же тип, что и x. Выходная дверь. |
ci | A Tensor. Должен иметь тот же тип, что и x. Вход ячейки. |
co | A Tensor. Должен иметь тот же тип, что и x. Ячейка после tanh. |
cs_grad | A Tensor. Должен иметь тот же тип, что и x. Текущий градиент cs. |
h_grad | A Tensor. Должен иметь тот же тип, что и x. Градиент вектора h. |
use_peephole | A bool. Использует ли ячейка связи с просматривающейся ячейкой. |
name | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
Кортеж объектов Tensor (cs_prev_grad, dicfo, wci_grad, wcf_grad, wco_grad). | |
cs_prev_grad | A Tensor. Имеет тот же тип, что и x. |
dicfo | A Tensor. Имеет тот же тип, что и x. |
wci_grad | A Tensor. Имеет тот же тип, что и x. |
wcf_grad | A Tensor. Имеет тот же тип, что и x. |
wco_grad | A Tensor. Имеет тот же тип, что и x. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/raw_ops/LSTMBlockCellGrad