Spec-Zone.ru › TensorFlow 2.4

tf.compat.v1.keras.layers.LSTMCell

Класс ячейки для слоя LSTM.

Наследуется от: Layer, Module

tf.compat.v1.keras.layers.LSTMCell(
    units, activation='tanh',
    recurrent_activation='hard_sigmoid', use_bias=True,
    kernel_initializer='glorot_uniform',
    recurrent_initializer='orthogonal',
    bias_initializer='zeros', unit_forget_bias=True,
    kernel_regularizer=None, recurrent_regularizer=None, bias_regularizer=None,
    kernel_constraint=None, recurrent_constraint=None, bias_constraint=None,
    dropout=0.0, recurrent_dropout=0.0, **kwargs
)
Аргументы
units Положительное целое число, размерность выходного пространства.
activation Функция активации для использования. По умолчанию: гиперболический тангенс (tanh). Если вы передадите None, активация не применяется (т.е. линейная активация: a(x) = x).
recurrent_activation Функция активации для использования на рекуррентном шаге. По умолчанию: жёсткая сигмоида (hard_sigmoid). Если вы передадите None, активация не применяется (т.е. линейная активация: a(x) = x).
use_bias Булево значение, использует ли слой вектор смещения.
kernel_initializer Инициализатор матрицы весов kernel, используемый для линейного преобразования входных данных.
recurrent_initializer Инициализатор матрицы весов recurrent_kernel, используемый для линейного преобразования рекуррентного состояния.
bias_initializer Инициализатор вектора смещения.
unit_forget_bias Булево значение. Если True, к смещению забывающего ворота добавляется 1 при инициализации. Установка в true также принудительно bias_initializer="zeros". Это рекомендуется в Jozefowicz et al., 2015
kernel_regularizer Функция регуляризации, применяемая к матрице весов kernel .
recurrent_regularizer Функция регуляризации, применяемая к матрице весов recurrent_kernel .
bias_regularizer Функция регуляризации, применяемая к вектору смещения.
kernel_constraint Функция ограничения, применяемая к матрице весов kernel .
recurrent_constraint Функция ограничения, применяемая к матрице весов recurrent_kernel .
bias_constraint Функция ограничения, применяемая к вектору смещения.
dropout Вещественное число от 0 до 1. Доля единиц, которые нужно отбросить для линейного преобразования входных данных.
recurrent_dropout Вещественное число от 0 до 1. Доля единиц, которые нужно отбросить для линейного преобразования рекуррентного состояния.

Аргументы вызова:

  • inputs: 2D тензор.
  • states: Список тензоров состояния, соответствующих предыдущему шагу во времени.
  • training: Булево значение Python, указывающее, должен ли слой вести себя в режиме обучения или в режиме вывода. Актуально только при использовании dropout или recurrent_dropout.

Методы

get_dropout_mask_for_cell

Просмотреть исходный код

get_dropout_mask_for_cell(
    inputs, training, count=1
)

Получить маску дропаута для входных данных ячейки RNN.

Создаст маску на основе контекста, если нет существующей кэшированной маски. Если создана новая маска, она обновит кэш в ячейке.

Аргументы
inputs Входной тензор, форма которого будет использована для генерации маски дропаута.
training Булевый тензор, находится ли он в режиме обучения, дропаут игнорируется в режиме без обучения.
count Целое число, сколько масок дропаута будет сгенерировано. Это полезно для ячеек, у которых внутренние веса объединены вместе.
Возвращает
Список тензоров масок, сгенерированных или кэшированных масок на основе контекста.

get_initial_state

Просмотреть исходный код

get_initial_state(
    inputs=None, batch_size=None, dtype=None
)

get_recurrent_dropout_mask_for_cell

Просмотреть исходный код

get_recurrent_dropout_mask_for_cell(
    inputs, training, count=1
)

Получить маску рекуррентного дропаута для ячейки RNN.

Создаст маску на основе контекста, если нет существующей кэшированной маски. Если создана новая маска, она обновит кэш в ячейке.

Аргументы
inputs Входной тензор, форма которого будет использована для генерации маски дропаута.
training Булевый тензор, находится ли он в режиме обучения, дропаут игнорируется в режиме без обучения.
count Целое число, сколько масок дропаута будет сгенерировано. Это полезно для ячеек, у которых внутренние веса объединены вместе.
Возвращает
Список тензоров масок, сгенерированных или кэшированных масок на основе контекста.

reset_dropout_mask

Просмотреть исходный код

reset_dropout_mask()

Сбросить кэшированные маски дропаута, если они есть.

Это важно для слоя RNN вызвать этот метод в методе call(), чтобы кэшированная маска была очищена перед вызовом cell.call(). Маска должна быть кэширована между шагами во времени в рамках одной пачки, но не должна быть кэширована между пачками. В противном случае это внесёт необоснованный перекос в определенные индексы данных в пачке.

reset_recurrent_dropout_mask

Просмотреть исходный код

reset_recurrent_dropout_mask()

Сбросить кэшированные маски рекуррентного дропаута, если они есть.

Это важно для слоя RNN вызвать этот метод в методе call(), чтобы кэшированная маска была очищена перед вызовом cell.call(). Маска должна быть кэширована между шагами во времени в рамках одной пачки, но не должна быть кэширована между пачками. В противном случае это внесёт необоснованный перекос в определенные индексы данных в пачке.

© 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/compat/v1/keras/layers/LSTMCell

Spec-Zone.ru

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