tf.raw_ops.GRUBlockCell

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

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

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

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

tf.compat.v1.raw_ops.GRUBlockCell

tf.raw_ops.GRUBlockCell(
    x, h_prev, w_ru, w_c, b_ru, b_c, name=None
)

Аргументы x: Вход в ячейку GRU. h_prev: Входное состояние из предыдущей ячейки GRU. w_ru: Матрица весов для ворот сброса и обновления. w_c: Матрица весов для ворот связи ячейки. b_ru: Вектор смещения для ворот сброса и обновления. b_c: Вектор смещения для ворот связи ячейки.

Возвращаемые значения r: Выход ворот сброса. u: Выход ворот обновления. c: Выход ворот связи ячейки. h: Текущее состояние ячейки GRU.

Примечание о обозначениях переменных:

Конкатенация a и b обозначается a_b. Элементное скалярное произведение a и b обозначается ab. Элементное скалярное произведение обозначается \circ. Матричное произведение обозначается *

Смещения инициализируются с помощью: b_ru - constant_initializer(1.0) b_c - constant_initializer(0.0)

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

x_h_prev = [x, h_prev]

[r_bar u_bar] = x_h_prev * w_ru + b_ru

r = sigmoid(r_bar)
u = sigmoid(u_bar)

h_prevr = h_prev \circ r

x_h_prevr = [x h_prevr]

c_bar = x_h_prevr * w_c + b_c
c = tanh(c_bar)

h = (1-u) \circ c + u \circ h_prev

Аргументы
`x`	A `Tensor`. Должен быть одним из следующих типов: `float32`.
`h_prev`	A `Tensor`. Должен иметь тот же тип, что и `x`.
`w_ru`	A `Tensor`. Должен иметь тот же тип, что и `x`.
`w_c`	A `Tensor`. Должен иметь тот же тип, что и `x`.
`b_ru`	A `Tensor`. Должен иметь тот же тип, что и `x`.
`b_c`	A `Tensor`. Должен иметь тот же тип, что и `x`.
`name`	Имя операции (необязательно).

Возвращаемые значения
Кортеж объектов `Tensor` (r, u, c, h).
`r`	A `Tensor`. Имеет тот же тип, что и `x`.
`u`	A `Tensor`. Имеет тот же тип, что и `x`.
`c`	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/GRUBlockCell