tf.keras.layers.CuDNNGRU
Быстрая реализация GRU, основанная на cuDNN.
tf.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 в пакете будет использоваться в качестве начального состояния для образца с индексом 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/CuDNNGRU