Spec-Zone.ru › TensorFlow 2.4

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. Может работать только на графическом процессоре.

Аргументы
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

Методы

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.4/api_docs/python/tf/compat/v1/keras/layers/CuDNNGRU

Spec-Zone.ru

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