Spec-Zone.ru › TensorFlow 2.4

tf.compat.v1.keras.layers.CuDNNLSTM

Быстрая реализация LSTM, основанная на cuDNN.

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

tf.compat.v1.keras.layers.CuDNNLSTM(
    units, kernel_initializer='glorot_uniform',
    recurrent_initializer='orthogonal',
    bias_initializer='zeros', unit_forget_bias=True,
    kernel_regularizer=None, recurrent_regularizer=None, bias_regularizer=None,
    activity_regularizer=None, kernel_constraint=None, recurrent_constraint=None,
    bias_constraint=None, return_sequences=False, return_state=False,
    go_backwards=False, stateful=False, **kwargs
)

Дополнительную информацию о cuDNN можно найти на веб-сайте разработчиков NVIDIA NVIDIA developer website. Может работать только на GPU.

Аргументы
units Положительное целое число, размерность выходного пространства.
kernel_initializer Инициализатор матрицы весов kernel, используемой для линейного преобразования входных данных.
unit_forget_bias Булево значение. Если True, к смещению забывающейся ячейки в момент инициализации добавляется 1. Установка в True также принудительно устанавливает bias_initializer="zeros". Это рекомендуется в Jozefowicz et al.
recurrent_initializer Инициализатор матрицы весов recurrent_kernel, используемой для линейного преобразования рекуррентного состояния.
bias_initializer Инициализатор вектора смещений.
kernel_regularizer Функция регуляризации, применяемая к матрице весов kernel.
recurrent_regularizer Функция регуляризации, применяемая к матрице весов recurrent_kernel.
bias_regularizer Функция регуляризации, применяемая к вектору смещений.
activity_regularizer Функция регуляризации, применяемая к выводу слоя (его "активации").
kernel_constraint Функция ограничения, применяемая к матрице весов kernel.
recurrent_constraint Функция ограничения, применяемая к матрице весов recurrent_kernel.
bias_constraint Функция ограничения, применяемая к вектору смещений.
return_sequences Булево значение. Возвращать ли последний выход в выходной последовательности или всю последовательность.
return_state Булево значение. Возвращать ли последнее состояние в дополнение к выходу.
go_backwards Булево значение (по умолчанию False). Если True, обрабатывать входную последовательность в обратном порядке и возвращать обратную последовательность.
stateful Булево значение (по умолчанию False). Если True, последнее состояние для каждого образца с индексом i в пакете будет использоваться в качестве начального состояния для образца с индексом i в следующем пакете.
Атрибуты
cell
states

Методы

reset_states

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

reset_states(
    states=None
)

Сбросить записанные состояния для состояния RNN-слоя.

Может использоваться только тогда, когда слой RNN создан с stateful = True. Args: states: Массивы NumPy, содержащие значение для начального состояния, которое будет передано ячейке в первый момент времени. Если значение None, будет создан нулевой массив NumPy на основе размера состояния ячейки.

Исключения
AttributeError Если слой RNN не является состоятельным.
ValueError Если размер пакета слоя RNN неизвестен.
ValueError Если входной массив NumPy несовместим с состоянием слоя RNN, либо по размеру, либо по типу данных.

© 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/CuDNNLSTM

Spec-Zone.ru

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