tf.keras.ops.extract_sequences

Расширяет размерность последнего осей на последовательности sequence_length.

tf.keras.ops.extract_sequences(
    x, sequence_length, sequence_stride
)

Проскальзывает окно размера sequence_length по последней оси входного сигнала с шагом sequence_stride, заменяя последнюю ось на [num_sequences, sequence_length] последовательностей.

Если размерность по последней оси равна N, количество последовательностей можно вычислить следующим образом:

num_sequences = 1 + (N - sequence_length) // sequence_stride

Пример:

x = keras.ops.convert_to_tensor([1, 2, 3, 4, 5, 6])
extract_sequences(x, 3, 2)
array([[1, 2, 3],
   [3, 4, 5]])