如何在张量流中使用api移位张量,就像nump.roll()或shift一样?

J. *_*ang 6 tensorflow

例如

换班:从[1, 2, 3][0, 1, 2][2, 3, 0]

对于滚动:从[1, 2, 3][2, 3, 1][3, 1, 2]

谢谢!