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