Spec-Zone.ru › TensorFlow 2.9

tf.compat.v1.keras.layers.CuDNNGRU

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

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

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 NVIDIA developer website. Может работать только на 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 в пакете будет использоваться в качестве начального состояния для образца с индексом i в следующем пакете.
Атрибуты
cell
states

Методы

get_losses_for

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

get_losses_for(
    inputs=None
)

reset_states

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

reset_states(
    states=None
)

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

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

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

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/compat/v1/keras/layers/CuDNNGRU

Spec-Zone.ru

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