tf.contrib.rnn.GRUBlockCell
Реализация блока ячейки GRU.
Наследуется от: LayerRNNCell
tf.contrib.rnn.GRUBlockCell(
num_units=None, cell_size=None, reuse=None, name='gru_cell'
)
Устарело: используйте GRUBlockCellV2 вместо этого.
Реализация основана на: http://arxiv.org/abs/1406.1078 Вычисляет прямое распространение ячейки GRU для 1 шага во времени.
Этот оператор ядра реализует следующие математические уравнения:
Смещения инициализируются следующим образом:
-
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
| Аргументы | |
|---|---|
num_units | int, Количество ячеек в ячейке GRU. |
cell_size | int, Старое (устаревшее) имя для num_units. |
reuse | (необязательно) булево значение, описывающее, следует ли повторно использовать переменные в существующем пространстве имен. Если не True, и в существующем пространстве имен уже есть заданные переменные, возникает ошибка. |
name | Строка, имя слоя. Слои с одинаковым именем будут использовать одни и те же веса, но для предотвращения ошибок в таких случаях требуется reuse=True. По умолчанию это "lstm_cell", для совместимости имён переменных с tf.compat.v1.nn.rnn_cell.GRUCell. |
| Исключения | |
|---|---|
ValueError | если и cell_size, и num_units не равны None; или оба равны None. |
| Атрибуты | |
|---|---|
graph | УСТАРЕВШАЯ ФУНКЦИЯ |
output_size | Целое число или TensorShape: размер выходных данных, производимых этой ячейкой. |
scope_name | |
state_size | Размер(ы) состояния(ий), используемых этой ячейкой. Он может быть представлен целым числом, TensorShape или кортежем целых чисел или TensorShape. |
Методы
get_initial_state
get_initial_state(
inputs=None, batch_size=None, dtype=None
)
zero_state
zero_state(
batch_size, dtype
)
Возвращает тензор(ы) состояния с нулевыми значениями.
| Аргументы | |
|---|---|
batch_size | int, float или тензор ячеек, представляющий размер пакета. |
dtype | тип данных для использования в состоянии. |
| Возвращаемые значения | |
|---|---|
Если state_size — это целое число или TensorShape, то возвращаемое значение — это тензор N-D формы [batch_size, state_size] заполненый нулями. Если |
© 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/r1.15/api_docs/python/tf/contrib/rnn/GRUBlockCell