Spec-Zone.ru › TensorFlow 2.9

tf.raw_ops.BlockLSTMV2

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

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

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

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

tf.compat.v1.raw_ops.BlockLSTMV2

tf.raw_ops.BlockLSTMV2(
    seq_len_max,
    x,
    cs_prev,
    h_prev,
    w,
    wci,
    wcf,
    wco,
    b,
    cell_clip=0,
    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)

Note that unlike LSTMBlockCell (and BlockLSTM) which uses ICFO gate layout,
this op uses IFCO. So in order for the following snippet to be equivalent
all gate-related outputs should be reordered.
Аргументы
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. Начальный вывод ячейки (для использования с peephole).
w A Tensor. Должен иметь тот же тип, что и x. Матрица весов.
wci A Tensor. Должен иметь тот же тип, что и x. Матрица весов для связи peephole входного шлюза.
wcf A Tensor. Должен иметь тот же тип, что и x. Матрица весов для связи peephole забывающего шлюза.
wco A Tensor. Должен иметь тот же тип, что и x. Матрица весов для связи peephole выходного шлюза.
b A Tensor. Должен иметь тот же тип, что и x. Вектор смещения.
cell_clip Необязательный float. По умолчанию 0. Значение для ограничения значения 'cs'.
use_peephole Необязательный bool. По умолчанию False. Используются ли веса peephole.
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/BlockLSTMV2

Spec-Zone.ru

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