tf.roll

Сдвигает элементы тензора вдоль оси.

tf.roll(
    input, shift, axis, name=None
)

Элементы сдвигаются положительно (в сторону больших индексов) на величину смещения shift вдоль размерности axis. Отрицательные shift значения сместят элементы в противоположном направлении. Элементы, которые пройдут за последнюю позицию, будут переходить на первую и наоборот. Можно указать несколько сдвигов вдоль нескольких осей.

Например:

# 't' is [0, 1, 2, 3, 4]
roll(t, shift=2, axis=0) ==> [3, 4, 0, 1, 2]

# shifting along multiple dimensions
# 't' is [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]]
roll(t, shift=[1, -2], axis=[0, 1]) ==> [[7, 8, 9, 5, 6], [2, 3, 4, 0, 1]]

# shifting along the same axis multiple times
# 't' is [[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]]
roll(t, shift=[2, -3], axis=[1, 1]) ==> [[1, 2, 3, 4, 0], [6, 7, 8, 9, 5]]