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