Spec-Zone.ru › TensorFlow 2.9

tf.raw_ops.BlockLSTM

Вычисляет прямое распространение ячейки LSTM для всех шагов во времени.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.raw_ops.BlockLSTM

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 A Tensor типа int64. Максимальная длина временного ряда, фактически используемая этим входом. Выводы дополняются нулями за пределами этой длины.
x A Tensor. Должен быть одним из следующих типов: half, float32. Последовательный вход в LSTM, форма (timelen, batch_size, num_inputs).
cs_prev A Tensor. Должен иметь тот же тип, что и x. Значение начального состояния ячейки.
h_prev A Tensor. Должен иметь тот же тип, что и x. Начальный вывод ячейки (для использования с просмотровыми значениями).
w A Tensor. Должен иметь тот же тип, что и x. Матрица весов.
wci A Tensor. Должен иметь тот же тип, что и x. Матрица весов для просмотровой связи входного шлюза.
wcf A Tensor. Должен иметь тот же тип, что и x. Матрица весов для просмотровой связи шлюза забывания.
wco A Tensor. Должен иметь тот же тип, что и x. Матрица весов для просмотровой связи выходного шлюза.
b A Tensor. Должен иметь тот же тип, что и x. Вектор смещения.
forget_bias Необязательный float. По умолчанию 1. Смещение шлюза забывания.
cell_clip Необязательный float. По умолчанию 3. Значение для ограничения значения 'cs'.
use_peephole Необязательный bool. По умолчанию False. Использовать ли просмотровые веса.
name Имя операции (необязательно).
Возвращаемое значение
Кортеж из объектов Tensor (i, cs, f, o, ci, co, h).
i A Tensor. Имеет тот же тип, что и x.
cs A Tensor. Имеет тот же тип, что и x.
f A Tensor. Имеет тот же тип, что и x.
o A Tensor. Имеет тот же тип, что и x.
ci A Tensor. Имеет тот же тип, что и x.
co A Tensor. Имеет тот же тип, что и x.
h 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/BlockLSTM

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API