tf.compat.v1.nn.bidirectional_dynamic_rnn
Создаёт динамическую версию двунаправленной рекуррентной нейронной сети. (устаревшее)
tf.compat.v1.nn.bidirectional_dynamic_rnn(
cell_fw,
cell_bw,
inputs,
sequence_length=None,
initial_state_fw=None,
initial_state_bw=None,
dtype=None,
parallel_iterations=None,
swap_memory=False,
time_major=False,
scope=None
)
Принимает входные данные и создаёт независимые прямые и обратные RNN. Размер входных данных для прямой и обратной ячейки должен совпадать. Начальное состояние для обоих направлений по умолчанию равно нулю (но может быть установлено необязательно), и промежуточные состояния никогда не возвращаются — сеть полностью развернута для заданной (переданной) длины(ы) последовательности(ей) или полностью развернута, если длина(ы) не задана.
| Аргументы | |
|---|---|
cell_fw | Экземпляр RNNCell, используемый для прямого направления. |
cell_bw | Экземпляр RNNCell, используемый для обратного направления. |
inputs | Входные данные RNN. Если time_major == False (по умолчанию), это должен быть тензор формы: [batch_size, max_time, ...], или вложенная кортеж таких элементов. Если time_major == True, это должен быть тензор формы: [max_time, batch_size, ...], или вложенная кортеж таких элементов. |
sequence_length | (необязательно) Вектор int32/int64, размер [batch_size], содержащий фактические длины для каждой из последовательностей в пакете. Если не предоставлено, предполагается, что все элементы пакета — полные последовательности; и обращение времени применяется от времени 0 до max_time для каждой последовательности. |
initial_state_fw | (необязательно) Начальное состояние для прямой RNN. Это должен быть тензор соответствующего типа и формы [batch_size, cell_fw.state_size]. Если cell_fw.state_size является кортежем, это должен быть кортеж тензоров с формами [batch_size, s] for s in cell_fw.state_size. |
initial_state_bw | (необязательно) То же, что и для initial_state_fw, но с использованием соответствующих свойств cell_bw. |
dtype | (необязательно) Тип данных для начальных состояний и ожидаемого выхода. Требуется, если начальные состояния не предоставлены или состояния RNN имеют разнородный тип данных. |
parallel_iterations | (по умолчанию: 32). Количество итераций, выполняемых параллельно. Те операции, которые не имеют временной зависимости и могут выполняться параллельно, будут. Этот параметр балансирует время и место. Значения >> 1 используют больше памяти, но занимают меньше времени, а меньшие значения используют меньше памяти, но вычисления занимают больше времени. |
swap_memory | Прозрачно меняет местоположение тензоров, созданных при прямом выводе, но необходимых для обратного распространения, с GPU на CPU. Это позволяет обучать RNN, которые обычно не помещаются на одном GPU, с очень минимальными (или без) потерями производительности. |
time_major | Формат формы тензоров inputs и outputs. Если true, эти Tensors должны иметь форму [max_time, batch_size, depth]. Если false, эти Tensors должны иметь форму [batch_size, max_time, depth]. Использование time_major = True немного эффективнее, потому что оно позволяет избежать транспонирования в начале и в конце вычислений RNN. Однако большинство данных TensorFlow имеют формат batch-major, поэтому по умолчанию эта функция принимает входные данные и выводит выходные данные в формате batch-major. |
scope | Область переменных для создаваемого подграфа; по умолчанию "bidirectional_rnn" |
| Возвращает | |
|---|---|
Кортеж (outputs, output_states), где: outputs: Кортеж (output_fw, output_bw), содержащий прямой и обратный выход RNN Tensor. Если time_major == False (по умолчанию), output_fw будет Tensor формы: [batch_size, max_time, cell_fw.output_size] и output_bw будет Tensor формы: [batch_size, max_time, cell_bw.output_size]. Если time_major == True, output_fw будет Tensor формы: [max_time, batch_size, cell_fw.output_size] и output_bw будет Tensor формы: [max_time, batch_size, cell_bw.output_size]. Он возвращает кортеж вместо единственного конкатенированного Tensor, в отличие от bidirectional_rnn. Если предпочтительнее конкатенированный, прямые и обратные выходы можно конкатенировать с помощью tf.concat(outputs, 2). output_states: Кортеж (output_state_fw, output_state_bw), содержащий конечные состояния прямого и обратного двунаправленного RNN. |
| Исключения | |
|---|---|
TypeError | Если cell_fw или cell_bw не является экземпляром RNNCell. |
© 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/compat/v1/nn/bidirectional_dynamic_rnn