Spec-Zone.ru › TensorFlow 1.15

tf.contrib.rnn.TimeReversedFusedRNN

Это адаптер для временного обращения ячейки FusedRNNCell.

Наследуется от: FusedRNNCell

tf.contrib.rnn.TimeReversedFusedRNN(
    cell
)

Например,

cell = tf.compat.v1.nn.rnn_cell.BasicRNNCell(10)
fw_lstm = tf.contrib.rnn.FusedRNNCellAdaptor(cell, use_dynamic_rnn=True)
bw_lstm = tf.contrib.rnn.TimeReversedFusedRNN(fw_lstm)
fw_out, fw_state = fw_lstm(inputs)
bw_out, bw_state = bw_lstm(inputs)

Методы

__call__

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

__call__(
    inputs, initial_state=None, dtype=None, sequence_length=None, scope=None
)

Запустить этот объединённый RNN на входных данных, начиная с заданного состояния.

Аргументы
inputs 3-D тензор с формой [time_len x batch_size x input_size] или список time_len тензоров с формой [batch_size x input_size].
initial_state либо тензор с формой [batch_size x state_size] , либо кортеж с формами [batch_size x s] for s in state_size, если ячейка принимает кортежи. Если это не указано, ожидается, что ячейка создаст начальное нулевое состояние типа dtype.
dtype Тип данных для начального состояния и ожидаемого результата. Требуется, если initial_state не указан или состояние RNN имеет разнородный тип.
sequence_length Указывает длину каждой последовательности ввода. Вектор (тензор) размера [batch_size], значения в [0, time_len). По умолчанию time_len для каждого элемента.
scope VariableScope или string для создаваемого подграфа; по умолчанию — имя класса.
Возвращаемое значение
Пара, содержащая:
  • Выход: 3-D тензор с формой [time_len x batch_size x output_size] или список time_len тензоров с формой [batch_size x output_size], чтобы соответствовать типу inputs.
  • Конечное состояние: либо один 2-D тензор, либо кортеж тензоров, соответствующий арности и формам initial_state.

© 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/contrib/rnn/TimeReversedFusedRNN

Spec-Zone.ru

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