這篇文章主要介紹了Tensorflow怎么獲取張量Tensor的具體維數,具有一定借鑒價值,感興趣的朋友可以參考下,希望大家閱讀完這篇文章之后大有收獲,下面讓小編帶著大家一起了解一下。
獲取Tensor的維數
>>> import tensorflow as tf >>> tf.__version__ '1.2.0-rc1' >>> x=tf.placeholder(dtype=float32,shape=[1,2,3,4]) >>> x=tf.placeholder(dtype=tf.float32,shape=[1,2,3,4]) >>> x.shape TensorShape([Dimension(1), Dimension(2), Dimension(3), Dimension(4)]) >>> x.get_shape() TensorShape([Dimension(1), Dimension(2), Dimension(3), Dimension(4)]) # 返回tuple >>> x.shape[2] Dimension(3) >>> x.get_shape()[2] Dimension(3) # 獲取具體維度數值 >>> x.shape[2].value 3 >>> x.get_shape()[2].value 3 # 也可以將TensorShape變量轉化為list類型,然后直接按照索引取值 >>> x.shape.as_list() [1, 2, 3, 4] >>> x.shape.as_list() [1, 2, 3, 4] # 可以與int型數值比較 >>> x.shape[2] == 3 True >>> x.get_shape()[2] == 3 True
感謝你能夠認真閱讀完這篇文章,希望小編分享的“Tensorflow怎么獲取張量Tensor的具體維數”這篇文章對大家有幫助,同時也希望大家多多支持億速云,關注億速云行業資訊頻道,更多相關知識等著你來學習!
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。