对张量或流张量进行Permute

Question

如何按照以下模式排列tf张量的特定维度：例如[0,1,2,3,4]->[1,2,3,4,0]？

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

Answer 1

试试这个：

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