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 для создаваемого подграфа; по умолчанию — имя класса. |
| Возвращаемое значение | |
|---|---|
Пара, содержащая:
|
© 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