tf.nn.bidirectional_dynamic_rnn
Создаёт динамическую версию двунаправленной рекуррентной нейронной сети. (устарело)
tf.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 — это данные по пакету, поэтому по умолчанию эта функция принимает входные данные и выдает выходные данные в формате пакета. |
scope | VariableScope для созданной подсети; по умолчанию "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. |
© 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/r1.15/api_docs/python/tf/nn/bidirectional_dynamic_rnn