Spec-Zone.ru › TensorFlow 2.3

tf.raw_ops.CudnnRNNBackpropV3

Обратное распространение ошибки для CudnnRNNV3.

Просмотр псевдонимов

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.raw_ops.CudnnRNNBackpropV3

tf.raw_ops.CudnnRNNBackpropV3(
    input, input_h, input_c, params, sequence_lengths, output, output_h, output_c,
    output_backprop, output_h_backprop, output_c_backprop, reserve_space,
    host_reserved, rnn_mode='lstm', input_mode='linear_input',
    direction='unidirectional', dropout=0, seed=0, seed2=0, num_proj=0,
    time_major=True, name=None
)

Вычисляет обратное распространение ошибки для данных и весов в RNN. Принимает дополнительный вход «sequence_lengths», в отличие от CudnnRNNBackprop.

rnn_mode: Указывает тип модели RNN. input_mode: Указывает, есть ли линейная проекция между входом и фактическим вычислением перед первым слоем. «skip_input» разрешено только при input_size == num_units; «auto_select» подразумевает «skip_input» при input_size == num_units; в противном случае подразумевает «linear_input». direction: Указывает, будет ли использоваться двунаправленная модель. Должно быть «unidirectional» или «bidirectional». dropout: Вероятность дропаута. При значении 0. дропаут отключен. seed: Первая часть начального значения для инициализации дропаута. seed2: Вторая часть начального значения для инициализации дропаута. input: Если time_major — True, это 3-мерный тензор с формой [seq_length, batch_size, input_size]. Если time_major — False, форма [batch_size, seq_length, input_size]. input_h: Если time_major — True, это 3-мерный тензор с формой [num_layer * dir, batch_size, num_units]. Если time_major — False, форма [batch_size, num_layer * dir, num_units]. input_c: Для LSTM, 3-мерный тензор с формой [num_layer * dir, batch, num_units]. Для других моделей он игнорируется. params: 1-мерный тензор, содержащий веса и смещения в непрозрачной структуре. Размер должен быть создан с помощью CudnnRNNParamsSize и инициализирован отдельно. Обратите внимание, что они могут быть несовместимы в разных поколениях. Поэтому рекомендуется сохранять и восстанавливать sequence_lengths: вектор длин каждой последовательности ввода. output: Если time_major — True, это 3-мерный тензор с формой [seq_length, batch_size, dir * num_units]. Если time_major — False, форма [batch_size, seq_length, dir * num_units]. output_h: Такая же форма, как у input_h. output_c: Такая же форма, как у input_c для LSTM. Пустой тензор для других моделей. output_backprop: 3-мерный тензор с такой же формой, как у output в прямом проходе. output_h_backprop: 3-мерный тензор с такой же формой, как у output_h в прямом проходе. output_c_backprop: 3-мерный тензор с такой же формой, как у output_c в прямом проходе. time_major: Указывает, является ли формат ввода/вывода временным или пакетным. reserve_space: Такое же значение reserve_space, которое было произведено в прямом проходе. input_backprop: Обратное распространение ошибки в вход в прямом проходе. Имеет такую же форму, как вход. input_h_backprop: Обратное распространение ошибки в input_h в прямом проходе. Имеет такую же форму, как input_h. input_c_backprop: Обратное распространение ошибки в input_c в прямом проходе. Имеет такую же форму, как input_c. params_backprop: Обратное распространение ошибки в буфер параметров в прямом проходе. Имеет такую же форму, как параметры.

Аргументы
input A Tensor. Должен быть одного из следующих типов: half, float32, float64.
input_h A Tensor. Должен иметь тот же тип, что и input.
input_c A Tensor. Должен иметь тот же тип, что и input.
params A Tensor. Должен иметь тот же тип, что и input.
sequence_lengths A Tensor типа int32.
output A Tensor. Должен иметь тот же тип, что и input.
output_h A Tensor. Должен иметь тот же тип, что и input.
output_c A Tensor. Должен иметь тот же тип, что и input.
output_backprop A Tensor. Должен иметь тот же тип, что и input.
output_h_backprop A Tensor. Должен иметь тот же тип, что и input.
output_c_backprop A Tensor. Должен иметь тот же тип, что и input.
reserve_space A Tensor. Должен иметь тот же тип, что и input.
host_reserved A Tensor типа int8.
rnn_mode Необязательный string из: "rnn_relu", "rnn_tanh", "lstm", "gru". По умолчанию "lstm".
input_mode Необязательный string из: "linear_input", "skip_input", "auto_select". По умолчанию "linear_input".
direction Необязательный string из: "unidirectional", "bidirectional". По умолчанию "unidirectional".
dropout Необязательный float. По умолчанию 0.
seed Необязательный int. По умолчанию 0.
seed2 Необязательный int. По умолчанию 0.
num_proj Необязательный int. По умолчанию 0.
time_major Необязательный bool. По умолчанию True.
name Имя операции (необязательно).
Возвращаемые значения
Кортеж объектов Tensor (input_backprop, input_h_backprop, input_c_backprop, params_backprop).
input_backprop A Tensor. Имеет тот же тип, что и input.
input_h_backprop A Tensor. Имеет тот же тип, что и input.
input_c_backprop A Tensor. Имеет тот же тип, что и input.
params_backprop A Tensor. Имеет тот же тип, что и input.

© 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/r2.3/api_docs/python/tf/raw_ops/CudnnRNNBackpropV3

Spec-Zone.ru

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