tf.transpose函数的用法

tf.transpose函数的用法

2017年06月16日 19:13:29 UESTC_C2_403 阅读数：24768更多

个人分类： tensorflow用法

tf.transpose(input, [dimension_1, dimenaion_2,..,dimension_n]):这个函数主要适用于交换输入张量的不同维度用的，如果输入张量是二维，就相当是转置。dimension_n是整数，如果张量是三维，就是用0,1,2来表示。这个列表里的每个数对应相应的维度。如果是[2,1,0]，就把输入张量的第三维度和第一维度交换。

例如：

1. import tensorflow as tf;

2. import numpy as np;

3.  

4. A = np.array([[1,2,3],[4,5,6]])

5. x = tf.transpose(A, [1,0])

6.  

7. B = np.array([[[1,2,3],[4,5,6]]])

8. y = tf.transpose(B, [2,1,0])

9. with tf.Session() as sess:

10. print A[1,0]

11. print sess.run(x[0,1])

12. print B[0,1,2]

13. print sess.run(y[2,1,0])

输出：

4
4
6
6