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`	`Tensor`. Должен иметь один из следующих типов: `float32`.
`h_prev`	`Tensor`. Должен иметь тот же тип, что и `x`.
`w_ru`	`Tensor`. Должен иметь тот же тип, что и `x`.
`w_c`	`Tensor`. Должен иметь тот же тип, что и `x`.
`b_ru`	`Tensor`. Должен иметь тот же тип, что и `x`.
`b_c`	`Tensor`. Должен иметь тот же тип, что и `x`.
`name`	Имя операции (необязательно).

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