tf.keras.layers.CuDNNLSTM

Быстрая реализация LSTM, поддерживаемая cuDNN.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.keras.layers.CuDNNLSTM

tf.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`

Методы

`get_initial_state`

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

get_initial_state(
    inputs
)

`reset_states`

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

reset_states(
    states=None
)

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