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