tf.contrib.rnn.LSTMBlockFusedCell
Реализация FusedRNNCell для LSTM.
Наследуется от: LSTMBlockWrapper
tf.contrib.rnn.LSTMBlockFusedCell(
num_units, forget_bias=1.0, cell_clip=None, use_peephole=False, reuse=None,
dtype=None, name='lstm_fused_cell'
)
Это крайне эффективная реализация LSTM, использующая одну операцию TF для всего LSTM. Она должна быть быстрее и эффективнее по памяти, чем LSTMBlockCell, определённая выше.
Реализация основана на: http://arxiv.org/abs/1409.2329
Мы добавляем forget_bias (по умолчанию: 1) к смещениям забывающего шлюза, чтобы уменьшить масштаб забывания в начале обучения.
Имена переменных согласованы с rnn_cell_impl.LSTMCell.
| Аргументы | |
|---|---|
num_units | целое число, количество ячеек в ячейке LSTM. |
forget_bias | число с плавающей точкой, смещение, добавляемое к забывающим шлюзам (см. выше). |
cell_clip | ограничить ячейку этим значением. По умолчанию ограничение отсутствует. |
use_peephole | Использовать ли соединения peephole. |
reuse | (необязательно) логическое значение, описывающее, следует ли повторно использовать переменные в существующем пространстве имен. Если не True, и существующее пространство имен уже содержит заданные переменные, возникает ошибка. |
dtype | тип данных переменных этого слоя. |
name | Строка, имя слоя. Слои с одинаковым именем будут разделять веса, но для избежания ошибок в таких случаях требуется reuse=True. По умолчанию это "lstm_cell", для совместимости имён переменных с tf.compat.v1.nn.rnn_cell.LSTMCell. |
| Атрибуты | |
|---|---|
graph | УСТАРЕВШАЯ ФУНКЦИЯ |
num_units | Количество ячеек в этой ячейке (размерность вывода). |
scope_name | |
© 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/LSTMBlockFusedCell
