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