Spec-Zone.ru › TensorFlow

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 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. Вектор смещений.
cell_clip Необязательный float. По умолчанию — 0. Значение, до которого ограничивается «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/BlockLSTMV2

Spec-Zone.ru

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