這篇文章主要介紹了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進行舉報，并提供相關證據，一經查實，將立刻刪除涉嫌侵權內容。
