tf.keras.ops.roll  |  TensorFlow v2.16.1 (original) (raw)

tf.keras.ops.roll

Roll tensor elements along a given axis.

View aliases

Main aliases

tf.keras.ops.numpy.roll

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

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

Args
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.
Returns
Output tensor.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates. Some content is licensed under the numpy license.

Last updated 2024-06-07 UTC.