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)>