Spec-Zone.ru › TensorFlow 2.3

tf.compat.v1.keras.layers.CuDNNGRU

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

tf.compat.v1.keras.layers.CuDNNGRU(
    units, kernel_initializer='glorot_uniform', recurrent_initializer='orthogonal',
    bias_initializer='zeros', 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. Может работать только на GPU.

Аргументы
units Положительное целое число, размерность выходного пространства.
kernel_initializer Инициализатор матрицы весов kernel, используемой для линейного преобразования входных данных.
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 в пакетe будет использоваться в качестве начального состояния для образца с индексом i в следующем пакетe.
Атрибуты
cell
states

Методы

reset_states

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

reset_states(
    states=None
)

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

Может быть использован только тогда, когда слой RNN построен с stateful = True. Аргументы: 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.3/api_docs/python/tf/compat/v1/keras/layers/CuDNNGRU

Spec-Zone.ru

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