tf.transpose()

import numpy as np
import tensorflow as tf

# x = np.array([[[1,2,3],
# [4,5,6]],
# [[7,8,9],
# [10,11,12]]])
#
# print(x.shape)

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

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

sess = tf.InteractiveSession()

print(sess.run(x))
print(sess.run(y))

>>>   [[1 2 3]
       [4 5 6]]

因為x是一個二維矩陣,
所以在tf.transpose里的第二個參數(shù)初始狀態(tài)應(yīng)該是[0,1]
那么[1,0]表示的就是第二維和第一維的數(shù)進行交換。


倘若:

import numpy as np
import tensorflow as tf

x = np.array([[[1,2,3],
               [4,5,6]],
              [[7,8,9],
               [10,11,12]]])

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

sess = tf.InteractiveSession()

print(sess.run(x))
print('\n\n')
print(sess.run(y))

>>>   [[[ 1 7]
        [ 4 10]]
       [[ 2 8]
        [ 5 11]]
       [[ 3 9]
        [ 6 12]]]

      [[[ 1 7]
        [ 4 10]]
       [[ 2 8]
        [ 5 11]]
       [[ 3 9]
        [ 6 12]]]

表示的就是將第3維和第1維的值進行交換。

?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結(jié)合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容