tf.raw_ops.BlockLSTMGrad
Вычисляет обратное распространение ошибки для ячейки LSTM по всей временной последовательности.
tf.raw_ops.BlockLSTMGrad(
seq_len_max,
x,
cs_prev,
h_prev,
w,
wci,
wcf,
wco,
b,
i,
cs,
f,
o,
ci,
co,
h,
cs_grad,
h_grad,
use_peephole,
name=None
)
Эта реализация предназначена для использования вместе с LSTMBlock.
| Аргументы | |
|---|---|
seq_len_max | Тензор Tensor типа int64. Максимальная длина временной последовательности, фактически используемая этим входом. Выходные данные дополняются нулями после этой длины. |
x | Тензор Tensor. Должен иметь один из следующих типов: half, float32. Входная последовательность для LSTM, форма (timelen, batch_size, num_inputs). |
cs_prev | Тензор Tensor. Должен иметь тот же тип, что и x. Начальное значение состояния ячейки. |
h_prev | Тензор Tensor. Должен иметь тот же тип, что и x. Начальный выход ячейки (для использования в связи peephole). |
w | Тензор Tensor. Должен иметь тот же тип, что и x. Матрица весов. |
wci | Тензор Tensor. Должен иметь тот же тип, что и x. Матрица весов для связи peephole со входным вентилем. |
wcf | Тензор Tensor. Должен иметь тот же тип, что и x. Матрица весов для связи peephole с вентилем забывания. |
wco | Тензор Tensor. Должен иметь тот же тип, что и x. Матрица весов для связи peephole с выходным вентилем. |
b | Тензор Tensor. Должен иметь тот же тип, что и x. Вектор смещения. |
i | Тензор Tensor. Должен иметь тот же тип, что и x. Входной вентиль на протяжении всей временной последовательности. |
cs | Тензор Tensor. Должен иметь тот же тип, что и x. Состояние ячейки до применения tanh на протяжении всей временной последовательности. |
f | Тензор Tensor. Должен иметь тот же тип, что и x. Вентиль забывания на протяжении всей временной последовательности. |
o | Тензор Tensor. Должен иметь тот же тип, что и x. Выходной вентиль на протяжении всей временной последовательности. |
ci | Тензор Tensor. Должен иметь тот же тип, что и x. Вход ячейки на протяжении всей временной последовательности. |
co | Тензор Tensor. Должен иметь тот же тип, что и x. Состояние ячейки после применения tanh на протяжении всей временной последовательности. |
h | Тензор Tensor. Должен иметь тот же тип, что и x. Вектор выходов h на протяжении всей временной последовательности. |
cs_grad | Тензор Tensor. Должен иметь тот же тип, что и x. Текущий градиент cs. |
h_grad | Тензор Tensor. Должен иметь тот же тип, что и x. Градиент вектора h. |
use_peephole | Тензор bool. Использовать ли веса peephole. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
Кортеж из объектов Tensor (x_grad, cs_prev_grad, h_prev_grad, w_grad, wci_grad, wcf_grad, wco_grad, b_grad). | |
x_grad | Тензор Tensor. Имеет тот же тип, что и x. |
cs_prev_grad | Тензор Tensor. Имеет тот же тип, что и x. |
h_prev_grad | Тензор Tensor. Имеет тот же тип, что и x. |
w_grad | Тензор Tensor. Имеет тот же тип, что и x. |
wci_grad | Тензор Tensor. Имеет тот же тип, что и x. |
wcf_grad | Тензор Tensor. Имеет тот же тип, что и x. |
wco_grad | Тензор Tensor. Имеет тот же тип, что и x. |
b_grad | Тензор 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/api_docs/python/tf/raw_ops/BlockLSTMGrad