tensorflow - 置换一个张量流张量

Question

如何在以下模式中置换我的 tf 张量的特定维度：例如 [0,1,2,3,4] --> [1,2,3,4,0]？

更一般地说，我如何旋转 n 个位置？

标签： tensorflow

Answer 1

尝试这个：

# 't' is [0, 1, 2, 3, 4]
tf.roll(t, shift=-1, axis=0) ==> [1, 2, 3, 4, 0]