tf.transpose函式的用法
阿新 • • 發佈:2018-12-17
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],就把輸入張量的第三維度和第一維度交換。
例如:
-
import tensorflow as tf;
-
import numpy as np;
-
A = np.array([[1,2,3],[4,5,6]])
-
x = tf.transpose(A, [1,0])
-
B = np.array([[[1,2,3],[4,5,6]]])
-
y = tf.transpose(B, [2,1,0])
-
with tf.Session() as sess:
-
print A[1,0]
-
print sess.run(x[0,1])
-
print B[0,1,2]
-
print sess.run(y[2,1,0])
輸出:
4
4
6
6