tf.keras.layers.SimpleRNNCell

Класс ячейки для SimpleRNN.

Наследуется от: Layer, Module

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,
    **kwargs
)

См. Руководство по API RNN Keras для получения подробной информации об использовании API RNN.

Этот класс обрабатывает один шаг в рамках всего входного временного ряда, в то время как tf.keras.layer.SimpleRNN обрабатывает весь ряд.

Аргументы вызова:

• inputs: 2D тензор со следующей формой: [batch, feature].
• states: 2D тензор со следующей формой: [batch, units], который является состоянием из предыдущего временного шага. Для нулевого временного шага будет использован начальное состояние, заданное пользователем.
• training: Булевое значение Python, указывающее, должен ли слой работать в режиме обучения или в режиме вывода. Актуально только при использовании dropout или recurrent_dropout.

Примеры:

inputs = np.random.random([32, 10, 8]).astype(np.float32)
rnn = tf.keras.layers.RNN(tf.keras.layers.SimpleRNNCell(4))

output = rnn(inputs)  # The output has shape `[32, 4]`.

rnn = tf.keras.layers.RNN(
    tf.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)

Методы

get_dropout_mask_for_cell

Просмотреть исходный код

get_dropout_mask_for_cell(
    inputs, training, count=1
)

Получить маску дропаута для входного тензора ячейки RNN.

Будет создана маска на основе контекста, если нет существующей кэшированной маски. Если новая маска сгенерирована, она обновит кэш в ячейке.

get_initial_state

Просмотреть исходный код

get_initial_state(
    inputs=None, batch_size=None, dtype=None
)

get_recurrent_dropout_mask_for_cell

Просмотреть исходный код

get_recurrent_dropout_mask_for_cell(
    inputs, training, count=1
)

Получить маску рекуррентного дропаута для ячейки RNN.

Будет создана маска на основе контекста, если нет существующей кэшированной маски. Если новая маска сгенерирована, она обновит кэш в ячейке.

reset_dropout_mask

Просмотреть исходный код

reset_dropout_mask()

Сбросить кэшированные маски дропаута, если таковые имеются.

Это важно для слоя RNN, чтобы вызвать этот метод в методе call(), чтобы кэшированная маска очистилась перед вызовом cell.call(). Маска должна кэшироваться на временном шаге внутри одной партии, но не должна кэшироваться между партиями. В противном случае это приведет к необоснованному смещению в отношении определенных индексов данных внутри партии.

reset_recurrent_dropout_mask

Просмотреть исходный код

reset_recurrent_dropout_mask()

Сбросить кэшированные маски рекуррентного дропаута, если таковые имеются.

Это важно для слоя RNN, чтобы вызвать этот метод в методе вызова (), чтобы кэшированная маска очистилась перед вызовом метода cell.call(). Маска должна кэшироваться на временном шаге внутри одной партии, но не должна кэшироваться между партиями. В противном случае это приведет к необоснованному смещению в отношении определенных индексов данных внутри партии.