tf.roll

t =[0, 1, 2, 3, 4] tf.roll(t, shift=2, axis=0) <tf.Tensor: shape=(5,), dtype=int32, numpy=array([3, 4, 0, 1, 2], dtype=int32)>

这里的 axis=[0]比较奇怪，0轴表示列，1轴表示行

t=[[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]] t1=tf.roll(t, shift=[1], axis=[0])print('t1=',t1) t2=tf.roll(t1, shift=[ -2], axis=[ 1]) print('t2=',t2) t1= tf.Tensor( [[5 6 7 8 9][0 1 2 3 4]], shape=(2, 5), dtype=int32) t2= tf.Tensor( [[7 8 9 5 6][2 3 4 0 1]], shape=(2, 5), dtype=int32)

可以把两部合并成一步执行

tf.roll(t, shift=[1, -2], axis=[0, 1]) ==> [[7, 8, 9, 5, 6], [2, 3, 4, 0, 1]]

这个函数在deepdream中用的比较多

《新程序员》：云原生和全面数字化实践50位技术专家共同创作，文字、视频、音频交互阅读

总结

以上是生活随笔为你收集整理的tf.roll的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。