tensorflow獲取變量維度信息
更新時間：2018年03月10日 08:45:04 作者：shuzfan
這篇文章主要為大家詳細介紹了tensorflow獲取變量維度信息，具有一定的參考價值，感興趣的小伙伴們可以參考一下
tensorflow版本1.4
獲取變量維度是一個使用頻繁的操作，在tensorflow中獲取變量維度主要用到的操作有以下三種：
- Tensor.shape
- Tensor.get_shape()
- tf.shape(input,name=None,out_type=tf.int32)
對上面三種操作做一下簡單分析：（這三種操作先記作A、B、C）
A 和 B 基本一樣，只不過前者是Tensor的屬性變量，后者是Tensor的函數(shù)。
A 和 B 均返回TensorShape類型，而 C 返回一個1D的out_type類型的Tensor。
A 和 B 可以在任意位置使用，而 C 必須在Session中使用。
A 和 B 獲取的是靜態(tài)shape，可以返回不完整的shape； C 獲取的是動態(tài)的shape，必須是完整的shape。
另外，補充從TenaorShape變量中獲取具體維度數(shù)值的方法
# 直接獲取TensorShape變量的第i個維度值 x.shape[i].value x.get_shape()[i].value # 將TensorShape變量轉(zhuǎn)化為list類型，然后直接按照索引取值 x.get_shape().as_list()
下面給出全部的示例程序：
import tensorflow as tf
x1 = tf.constant([[1,2,3],[4,5,6]])
# 占位符創(chuàng)建變量，第一個維度初始化為None，表示暫不指定維度
x2 = tf.placeholder(tf.float32,[None, 2,3])
print('x1.shape:',x1.shape)
print('x2.shape:',x2.shape)
print('x2.shape[1].value:',x2.shape[1].value)
print('tf.shape(x1):',tf.shape(x1))
print('tf.shape(x2):',tf.shape(x2))
print('x1.get_shape():',x1.get_shape())
print('x2.get_shape():',x2.get_shape())
print('x2.get_shape.as_list[1]:',x2.get_shape().as_list()[1])
shapeOP1 = tf.shape(x1)
shapeOP2 = tf.shape(x2)
with tf.Session() as sess:
print('Within session, tf.shape(x1):',sess.run(shapeOP1))
# 由于x2未進行完整的變量填充，其維度不完整，因此執(zhí)行下面的命令將會報錯
# print('Within session, tf.shape(x2):',sess.run(shapeOP2)) # 此命令將會報錯
輸出結(jié)果為：
x1.shape: (2, 3)
x2.shape: (?, 2, 3)
x2.shape[1].value: 2
tf.shape(x1): Tensor("Shape:0", shape=(2,), dtype=int32)
tf.shape(x2): Tensor("Shape_1:0", shape=(3,), dtype=int32)
x1.get_shape(): (2, 3)
x2.get_shape(): (?, 2, 3)
x2.get_shape.as_list[1]: 2
Within session, tf.shape(x1): [2 3]
以上就是本文的全部內(nèi)容，希望對大家的學(xué)習有所幫助，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要介紹了python flask框架實現(xiàn)重定向功能,結(jié)合實例形式分析了flask框架重定向功能的實現(xiàn)與使用方法,需要的朋友可以參考下2019-07-07
這篇文章主要為大家詳細介紹了tensorflow實現(xiàn)KNN識別MNIST，具有一定的參考價值，感興趣的小伙伴們可以參考一下2018-03-03
這篇文章主要是帶大家寫一個利用Turtle庫制作的一款海龜闖關(guān)的三大迷宮，文中的示例代碼講解詳細，對我們學(xué)習Python有一定幫助，感興趣的可以了解一下2022-04-04
python中使用ctypes調(diào)用so傳參設(shè)置遇到的問題及解決方法
這篇文章主要介紹了python中使用ctypes調(diào)用so傳參設(shè)置，本文較詳細的給大家介紹了遇到問題及解決方案,需要的朋友可以參考下2019-06-06
這篇文章主要介紹了Flask框架中request、請求鉤子、上下文用法,結(jié)合實例形式分析了flask框架中request、請求鉤子及上下文的功能、用法及相關(guān)操作注意事項,需要的朋友可以參考下2019-07-07
電腦網(wǎng)絡(luò)有點問題,直接安裝第三方庫的時候出現(xiàn)了各種問題,所以在這篇文章里記錄一下各種安裝第三方庫的方法吧,下面這篇文章主要給大家介紹了關(guān)于Python安裝第三方庫常用方法的相關(guān)資料,需要的朋友可以參考下2022-04-04
這篇文章主要介紹了速記Python布爾值，具有一定參考價值，需要的朋友可以了解下。2017-11-11
python函數(shù)中將變量名轉(zhuǎn)換成字符串實例
這篇文章主要介紹了python函數(shù)中將變量名轉(zhuǎn)換成字符串實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-05-05
使用python查找替換PowerPoint演示文稿中的文本
演示文稿已成為商務(wù)會議、學(xué)術(shù)報告和教育培訓(xùn)中不可或缺的一部分,而PowerPoint演示文稿作為行業(yè)標準工具,更是承載著無數(shù)創(chuàng)意與信息的載體,本文將介紹如何使用Python來精確查找并替換PowerPoint演示文稿中的文本,需要的朋友可以參考下2024-07-07
分享Pandas庫中的一些寶藏函數(shù)transform()
Pandas具有很多強大的功能，transform就是其中之一，利用它可以高效地匯總數(shù)據(jù)且不改變數(shù)據(jù)行數(shù)，transform是一種什么數(shù)據(jù)操作？如果熟悉SQL的窗口函數(shù)，就非常容易理解了2021-09-09