tf.raw_ops.LSTMBlockCell

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

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

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

Дополнительные сведения см. в руководстве по миграции.

tf.compat.v1.raw_ops.LSTMBlockCell

tf.raw_ops.LSTMBlockCell(
    x,
    cs_prev,
    h_prev,
    w,
    wci,
    wcf,
    wco,
    b,
    forget_bias=1,
    cell_clip=3,
    use_peephole=False,
    name=None
)

В этой реализации используется 1 матрица весов и 1 вектор смещений, и есть необязательное соединение peephole.

Этот оператор ядра реализует следующие математические уравнения:

xh = [x, h_prev]
[i, f, ci, o] = xh * w + b
f = f + forget_bias

if not use_peephole:
  wci = wcf = wco = 0

i = sigmoid(cs_prev * wci + i)
f = sigmoid(cs_prev * wcf + f)
ci = tanh(ci)

cs = ci .* i + cs_prev .* f
cs = clip(cs, cell_clip)

o = sigmoid(cs * wco + o)
co = tanh(cs)
h = co .* o

Аргументы
`x`	A `Tensor`. Должен быть одним из следующих типов: `half`, `float32`. Вход в ячейку LSTM, форма (размер_пакета, количество_входов).
`cs_prev`	A `Tensor`. Должен иметь тот же тип, что и `x`. Значение состояния ячейки на предыдущем шаге времени.
`h_prev`	A `Tensor`. Должен иметь тот же тип, что и `x`. Выход предыдущей ячейки на предыдущем шаге времени.
`w`	A `Tensor`. Должен иметь тот же тип, что и `x`. Матрица весов.
`wci`	A `Tensor`. Должен иметь тот же тип, что и `x`. Матрица весов для соединения peephole входного шлюза.
`wcf`	A `Tensor`. Должен иметь тот же тип, что и `x`. Матрица весов для соединения peephole забывающего шлюза.
`wco`	A `Tensor`. Должен иметь тот же тип, что и `x`. Матрица весов для соединения peephole выходного шлюза.
`b`	A `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`	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/LSTMBlockCell