tf.raw_ops.BlockLSTM
Вычисляет прямое распространение сигнала через ячейку LSTM для всех временных шагов.
tf.raw_ops.BlockLSTM(
seq_len_max,
x,
cs_prev,
h_prev,
w,
wci,
wcf,
wco,
b,
forget_bias=1,
cell_clip=3,
use_peephole=False,
name=None
)
Это эквивалентно применению LSTMBlockCell в цикле, например так:
for x1 in unpack(x):
i1, cs1, f1, o1, ci1, co1, h1 = LSTMBlock(
x1, cs_prev, h_prev, w, wci, wcf, wco, b)
cs_prev = cs1
h_prev = h1
i.append(i1)
cs.append(cs1)
f.append(f1)
o.append(o1)
ci.append(ci1)
co.append(co1)
h.append(h1)
return pack(i), pack(cs), pack(f), pack(o), pack(ci), pack(ch), pack(h)
| Аргументы | |
|---|---|
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. Вектор смещений. |
forget_bias | Необязательный float. Значение по умолчанию — 1. Смещение вентиля забывания. |
cell_clip | Необязательный float. Значение по умолчанию — 3. Значение, до которого следует ограничить «cs». |
use_peephole | Необязательный bool. Значение по умолчанию — False. Использовать ли веса peephole. |
name | Имя операции (необязательно). |
| Возвращаемые значения | |
|---|---|
Кортеж объектов Tensor (i, cs, f, o, ci, co, h). | |
i | Tensor. Имеет тот же тип, что и x. |
cs | Tensor. Имеет тот же тип, что и x. |
f | Tensor. Имеет тот же тип, что и x. |
o | Tensor. Имеет тот же тип, что и x. |
ci | Tensor. Имеет тот же тип, что и x. |
co | Tensor. Имеет тот же тип, что и x. |
h | 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/BlockLSTM