1. 函数说明

先说结论：
tf.roll(t, shift=[1, -2], axis=[0, 1])：
shift：正值表示右（或下）移，负值表示左（或上）移
axis：tensor不只一行时：0轴表示列，1轴表示行；只有一行时，0，1都是按行
表示整体沿着(axis=0)列下移1位，再沿着(axis=1)行方向右移2位

情况1:axis=0，但是tensor只有一行

tf.roll(t, shift=2, axis=0)：
shift：正值表示右移，负值表示左移，这里表示右移2位，
axis=0，按行移动

情况2: axis=0或axis=[0]，tensor不只一行

t1=tf.roll(t, shift=[1], axis=[0])
axis=0, axis=[0]：tensor不只一行，0轴表示列，1轴表示行(有括号是不一样的)

情况3:axis=[0, 1]

tf.roll(t, shift=[1, -2], axis=[0, 1]) ：
表示整体沿着列下移1位，再沿着行方向右移2位

情况4:不加shift和axis

tf.roll(t, [1, -2], [0, 1]) ：
和情况3一样

2. 举例说明

例1:

输入：

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)>

例2和例3:

输入:

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)

例4:可以把例2和例3两部合并成一步执行:

输入：

tf.roll(t, shift=[1, -2], axis=[0, 1])

输出：

t2= tf.Tensor(
[[7 8 9 5 6][2 3 4 0 1]], shape=(2, 5), dtype=int32)

tf.roll：tensorflow 中对多维Tensor移位相关推荐

TensorFlow中Session.run和Tensor.eval的区别
之前在TensorFlow中运行代码时,在会话中会需要运行节点,会碰到两种方式:Session.run()和Tensor.eval(),刚开始不太懂这两者之间的差异,最后通过查找官方文档和一些资料了解 ...
tensorflow 中，修改张量tensor特定元素的值
tensorflow中: constant tensor不能直接赋值,否则会报错: TypeError: 'Tensor' object does not support ...
tensorflow中张量、常量、变量、占位符
引言从实例出发 #先导入TensorFlow import tensorflow as tf# Create TensorFlow object called hello_constant hell ...
TensorFlow实战：TensorFlow中的CNN
这里按照官方api介绍官方api点这里卷积不同的ops下使用的卷积操作总结如下: conv2d:Arbitrary filters that can mix channels together(通 ...
Tensorflow中的tf.argmax()函数
转载出处:http://www.cnblogs.com/willnote/p/6758953.html 官方API定义 tf.argmax(input, axis=None, name=None, d ...
tensorflow中的tf.summary.image
tensorflow中的tf.summary.image tf.summary.image(name,#生成的节点的名称.也将作为TensorBoard中的系列名称tensor,#uint8或者flo ...
tensorflow中打印一个tensor值
tensorflow中打印一个tensor值代码: import tensorflow as tf x = tf.ones(shape=[2, 3], dtype=tf.int32,name='x' ...
tensorflow中tensor的索引
tensorflow中tensor的索引 1.print(sess.run(outputs[0:2,0:2,:])) 2.print(sess.run(tf.slice(outputs,[0,0,0] ...
tensorflow从入门到精通100讲（一）-如何申明一个tensor，Tensorflow中最重要的基本操作
前言如果把Tensorflow翻译成中文,意思是流动的张量, 第零阶张量为标量,第一阶张量为矢量, 第二阶张量则成为矩阵,在tensorflow中流动的数据大多都以这三种数据结构流动,那么如何去申明 ...

tf.roll：tensorflow 中对多维Tensor移位

1. 函数说明

2. 举例说明

tf.roll：tensorflow 中对多维Tensor移位相关推荐

最新文章

热门文章