tf.keras.layers.GRUCell
Класс ячейки для слоя GRU.
Наследуется от: Layer, Operation
tf.keras.layers.GRUCell(
units,
activation='tanh',
recurrent_activation='sigmoid',
use_bias=True,
kernel_initializer='glorot_uniform',
recurrent_initializer='orthogonal',
bias_initializer='zeros',
kernel_regularizer=None,
recurrent_regularizer=None,
bias_regularizer=None,
kernel_constraint=None,
recurrent_constraint=None,
bias_constraint=None,
dropout=0.0,
recurrent_dropout=0.0,
reset_after=True,
seed=None,
**kwargs
)
Этот класс обрабатывает один шаг последовательности входных данных, тогда как keras.layer.GRU обрабатывает всю последовательность.
| Аргументы | |
|---|---|
units | Положительное целое число, размерность пространства выходных данных. |
activation | Используемая функция активации. По умолчанию: гиперболический тангенс (tanh). Если передать None, активация не применяется (то есть используется линейная активация: a(x) = x). |
recurrent_activation | Функция активации для рекуррентного шага. По умолчанию: сигмоида (sigmoid). Если передать None, активация не применяется (то есть используется линейная активация: a(x) = x). |
use_bias | Логическое значение (по умолчанию True), определяющее, должен ли слой использовать вектор смещения. |
kernel_initializer | Инициализатор матрицы весов kernel, используемой для линейного преобразования входных данных. По умолчанию: "glorot_uniform". |
recurrent_initializer | Инициализатор матрицы весов recurrent_kernel, используемой для линейного преобразования рекуррентного состояния. По умолчанию: "orthogonal". |
bias_initializer | Инициализатор вектора смещения. По умолчанию: "zeros". |
kernel_regularizer | Функция регуляризации, применяемая к матрице весов kernel. По умолчанию: None. |
recurrent_regularizer | Функция регуляризации, применяемая к матрице весов recurrent_kernel. По умолчанию: None. |
bias_regularizer | Функция регуляризации, применяемая к вектору смещения. По умолчанию: None. |
kernel_constraint | Функция ограничения, применяемая к матрице весов kernel. По умолчанию: None. |
recurrent_constraint | Функция ограничения, применяемая к матрице весов recurrent_kernel. По умолчанию: None. |
bias_constraint | Функция ограничения, применяемая к вектору смещения. По умолчанию: None. |
dropout | Число с плавающей точкой от 0 до 1. Доля блоков, отбрасываемых при линейном преобразовании входных данных. По умолчанию: 0. |
recurrent_dropout | Число с плавающей точкой от 0 до 1. Доля блоков, отбрасываемых при линейном преобразовании рекуррентного состояния. По умолчанию: 0. |
reset_after | Соглашение GRU (применять ли вентиль сброса до или после умножения матриц). False = «до», True = «после» (по умолчанию и совместимо с cuDNN). |
seed | Начальное значение генератора случайных чисел для dropout. |
| Аргументы вызова | |
|---|---|
inputs | Тензор размерности 2 с формой (batch, features). |
states | Тензор размерности 2 с формой (batch, units), представляющий состояние на предыдущем временном шаге. |
training | Логическое значение Python, указывающее, должен ли слой работать в режиме обучения или в режиме вывода. Имеет значение, только если используется dropout или recurrent_dropout. |
Пример:
inputs = np.random.random((32, 10, 8)) rnn = keras.layers.RNN(keras.layers.GRUCell(4)) output = rnn(inputs) output.shape (32, 4) rnn = keras.layers.RNN( keras.layers.GRUCell(4), return_sequences=True, return_state=True) whole_sequence_output, final_state = rnn(inputs) whole_sequence_output.shape (32, 10, 4) final_state.shape (32, 4)
| Атрибуты | |
|---|---|
input | Возвращает входной тензор или тензоры символьной операции. Возвращает только тензор или тензоры, соответствующие первому вызову операции. |
output | Возвращает выходной тензор или тензоры слоя. Возвращает только тензор или тензоры, соответствующие первому вызову операции. |
Методы
from_config
@classmethod
from_config(
config
)
Создает слой на основе его конфигурации.
Этот метод является обратным к get_config и позволяет создать тот же слой из словаря конфигурации. Он не обрабатывает связи между слоями (это делает Network) и веса (это делает set_weights).
| Аргументы | |
|---|---|
config | Словарь Python, обычно являющийся результатом get_config. |
| Возвращает | |
|---|---|
| Экземпляр слоя. |
get_dropout_mask
get_dropout_mask(
step_input
)
get_initial_state
get_initial_state(
batch_size=None
)
get_recurrent_dropout_mask
get_recurrent_dropout_mask(
step_input
)
reset_dropout_mask
reset_dropout_mask()
Сбрасывает кэшированную маску dropout, если она есть.
Слой RNN вызывает этот метод в call(), чтобы очистить кэшированную маску после вызова cell.call(). Маска должна кэшироваться на всех временных шагах в пределах одного пакета, но не между пакетами.
reset_recurrent_dropout_mask
reset_recurrent_dropout_mask()
symbolic_call
symbolic_call(
*args, **kwargs
)
© 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/api_docs/python/tf/keras/layers/GRUCell