Spec-Zone.ru › TensorFlow 2.4

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 вектор смещений, и есть необязательная петельчатая связь.

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

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. Матрица весов для петельчатой связи входного входа.
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.

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/raw_ops/LSTMBlockCell

Spec-Zone.ru

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