tensorflow 中对向量的的单步移位（错位）：tf.roll

最新推荐文章于 2023-04-20 15:35:00 发布

原创最新推荐文章于 2023-04-20 15:35:00 发布 · 973 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #人工智能 #tensorflow

tensorflow 专栏收录该内容

1 篇文章

订阅专栏

本文介绍了TensorFlow中tf.roll函数的使用方法及其参数设置。该函数可以在指定轴上实现张量元素的循环移位操作，适用于多维数据处理。文章通过示例展示了不同情况下tf.roll函数的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

原来想要是先这个变换，查了一下tensorflow还真有，但是看上去很少有人用所以记录一下

TensorFlow Core v2.3.0

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

在tf1里是

tf.manip.roll(
    input,
    shift,
    axis
)

示例

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