tf.reverse_sequence

Обращает переменные фрагменты.

tf.reverse_sequence(
    input, seq_lengths, seq_axis=None, batch_axis=None, name=None
)

Данный оператор вначале делит input вдоль измерения batch_axis, и для каждого фрагмента i, инвертирует первые seq_lengths[i] элементы вдоль измерения seq_axis.

Элементы seq_lengths должны подчиняться seq_lengths[i] <= input.dims[seq_axis], и seq_lengths должно быть вектором длиной input.dims[batch_axis].

Фрагмент вывода i вдоль измерения batch_axis затем получается из фрагмента входных данных i, с первыми seq_lengths[i] фрагментами вдоль измерения seq_axis, которые были инвертированы.

Пример использования:

seq_lengths = [7, 2, 3, 5]
input = [[1, 2, 3, 4, 5, 0, 0, 0], [1, 2, 0, 0, 0, 0, 0, 0],
         [1, 2, 3, 4, 0, 0, 0, 0], [1, 2, 3, 4, 5, 6, 7, 8]]
output = tf.reverse_sequence(input, seq_lengths, seq_axis=1, batch_axis=0)
output
<tf.Tensor: shape=(4, 8), dtype=int32, numpy=
array([[0, 0, 5, 4, 3, 2, 1, 0],
       [2, 1, 0, 0, 0, 0, 0, 0],
       [3, 2, 1, 4, 0, 0, 0, 0],
       [5, 4, 3, 2, 1, 6, 7, 8]], dtype=int32)>