tf.keras.layers.SimpleRNNCell
Класс ячейки для SimpleRNN.
Наследует от: Layer, Operation
tf.keras.layers.SimpleRNNCell(
units,
activation='tanh',
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,
seed=None,
**kwargs
)
Этот класс обрабатывает один шаг всей входной временной последовательности, тогда как keras.layer.SimpleRNN обрабатывает всю последовательность.
| Аргументы | |
|---|---|
units | Положительное целое число, размерность выходного пространства. |
activation | Используемая функция активации. По умолчанию: гиперболический тангенс (tanh). Если передать 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. |
seed | Случайное начальное значение для dropout. |
| Аргументы вызова | |
|---|---|
sequence | Двумерный тензор формы (batch, features). |
states | Двумерный тензор формы (batch, units), представляющий состояние на предыдущем временном шаге. |
training | Логическое значение Python, указывающее, должен ли слой работать в режиме обучения или в режиме вывода. Имеет значение только при использовании dropout или recurrent_dropout. |
Пример:
inputs = np.random.random([32, 10, 8]).astype(np.float32)
rnn = keras.layers.RNN(keras.layers.SimpleRNNCell(4))
output = rnn(inputs) # The output has shape `(32, 4)`.
rnn = keras.layers.RNN(
keras.layers.SimpleRNNCell(4),
return_sequences=True,
return_state=True
)
# whole_sequence_output has shape `(32, 10, 4)`.
# final_state has shape `(32, 4)`.
whole_sequence_output, final_state = rnn(inputs)
| Атрибуты | |
|---|---|
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/SimpleRNNCell