tf.keras.ops.roll

Roll tensor elements along a given axis.

Main aliases

tf.keras.ops.numpy.roll

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.