tf.raw_ops.CudnnRNNBackpropV3
Обратная проходка шага CudnnRNNV3.
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: Указывает, будет ли использоваться двунаправленная модель. Должно быть «однонаправленное» или «двунаправленное». dropout: Вероятность дропаута. При значении 0. дропаут отключён. seed: Первая часть семени для инициализации дропаута. seed2: Вторая часть семени для инициализации дропаута. input: Если time_major истинно, это 3-мерный тензор с формой [seq_length, batch_size, input_size]. Если time_major ложно, форма [batch_size, seq_length, input_size]. input_h: Если time_major истинно, это 3-мерный тензор с формой [num_layer * dir, batch_size, num_units]. Если time_major ложно, форма [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 истинно, это 3-мерный тензор с формой [seq_length, batch_size, dir * num_units]. Если time_major ложно, форма [batch_size, seq_length, dir * num_units]. output_h: Такая же форма, как input_h. output_c: Такая же форма, как input_c для LSTM. Пустой тензор для других моделей. output_backprop: 3-мерный тензор с такой же формой, как выход в прямом проходе. 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 | А Tensor. Должен быть одним из следующих типов: half, float32, float64. |
input_h | А Tensor. Должен иметь тот же тип, что и input. |
input_c | А Tensor. Должен иметь тот же тип, что и input. |
params | А Tensor. Должен иметь тот же тип, что и input. |
sequence_lengths | А Tensor типа int32. |
output | А Tensor. Должен иметь тот же тип, что и input. |
output_h | А Tensor. Должен иметь тот же тип, что и input. |
output_c | А Tensor. Должен иметь тот же тип, что и input. |
output_backprop | А Tensor. Должен иметь тот же тип, что и input. |
output_h_backprop | А Tensor. Должен иметь тот же тип, что и input. |
output_c_backprop | А Tensor. Должен иметь тот же тип, что и input. |
reserve_space | А Tensor. Должен иметь тот же тип, что и input. |
host_reserved | А 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 | А Tensor. Имеет тот же тип, что и input. |
input_h_backprop | А Tensor. Имеет тот же тип, что и input. |
input_c_backprop | А Tensor. Имеет тот же тип, что и input. |
params_backprop | А Tensor. Имеет тот же тип, что и input. |
© 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/versions/r2.9/api_docs/python/tf/raw_ops/CudnnRNNBackpropV3