Spec-Zone.ru › TensorFlow

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

Spec-Zone.ru

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