tf.keras.ops.roll

Roll tensor elements along a given axis.

tf.keras.ops.roll(
    x, shift, axis=None
)

Elements that roll beyond the last position are re-introduced at the first.

x Input tensor.
shift The number of places by which elements are shifted.
axis The axis along which elements are shifted. By default, the array is flattened before shifting, after which the original shape is restored.

Output tensor.