tensorflow2.0的函數(shù)簽名與圖結(jié)構(gòu)(推薦)
input_signature的好處:
1.可以限定函數(shù)的輸入類型,以防止調(diào)用函數(shù)時(shí)調(diào)錯(cuò),
2.一個(gè)函數(shù)有了input_signature之后,在tensorflow里邊才可以保存成savedmodel。在保存成savedmodel的過程中,需要使用get_concrete_function函數(shù)把一個(gè)tf.function標(biāo)注的普通的python函數(shù)變成帶有圖定義的函數(shù)。
下面的代碼具體體現(xiàn)了input_signature可以限定函數(shù)的輸入類型這一作用。
@tf.function(input_signature=[tf.TensorSpec([None], tf.int32, name='x')]) def cube(z): #實(shí)現(xiàn)輸入的立方 return tf.pow(z, 3) try: print(cube(tf.constant([1., 2., 3.]))) except ValueError as ex: print(ex) print(cube(tf.constant([1, 2, 3])))
輸出:
Python inputs incompatible with input_signature:
inputs: (
tf.Tensor([1. 2. 3.], shape=(3,), dtype=float32))
input_signature: (
TensorSpec(shape=(None,), dtype=tf.int32, name='x'))
tf.Tensor([ 1 8 27], shape=(3,), dtype=int32)
get_concrete_function的使用
note:首先說明,下面介紹的函數(shù)在模型構(gòu)建、模型訓(xùn)練的過程中不會(huì)用到,下面介紹的函數(shù)主要用在兩個(gè)地方:1、如何保存模型 2、保存好模型后,如何載入進(jìn)來。
可以給 由@tf.function標(biāo)注的普通的python函數(shù),給它加上input_signature, 從而讓這個(gè)python函數(shù)變成一個(gè)可以保存的tensorflow圖結(jié)構(gòu)(SavedModel)
舉例說明函數(shù)的用法:
@tf.function(input_signature=[tf.TensorSpec([None], tf.int32, name='x')]) def cube(z): return tf.pow(z, 3) try: print(cube(tf.constant([1., 2., 3.]))) except ValueError as ex: print(ex) print(cube(tf.constant([1, 2, 3]))) # @tf.function py func -> tf graph # get_concrete_function -> add input signature -> SavedModel cube_func_int32 = cube.get_concrete_function( tf.TensorSpec([None], tf.int32)) #tensorflow的類型 print(cube_func_int32)
輸出:
<tensorflow.python.eager.function.ConcreteFunction object at 0x00000240E29695C0>
從輸出結(jié)果可以看到:調(diào)用get_concrete_function函數(shù)后,輸出的是一個(gè)ConcreteFunction對象
#看用新參數(shù)獲得的對象與原來的對象是否一樣 print(cube_func_int32 is cube.get_concrete_function( tf.TensorSpec([5], tf.int32))) #輸入大小為5 print(cube_func_int32 is cube.get_concrete_function( tf.constant([1, 2, 3]))) #傳具體數(shù)據(jù)
輸出:
True
True
cube_func_int32.graph #圖定義
輸出:
[<tf.Operation 'x' type=Placeholder>, <tf.Operation 'Pow/y' type=Const>, <tf.Operation 'Pow' type=Pow>, <tf.Operation 'Identity' type=Identity>]
pow_op = cube_func_int32.graph.get_operations()[2] print(pow_op)
輸出:
name: "Pow"
op: "Pow"
input: "x"
input: "Pow/y"
attr {
key: "T"
value {
type: DT_INT32
}
}
print(list(pow_op.inputs)) print(list(pow_op.outputs))
輸出:
[<tf.Tensor 'x:0' shape=(None,) dtype=int32>, <tf.Tensor 'Pow/y:0' shape=() dtype=int32>]
[<tf.Tensor 'Pow:0' shape=(None,) dtype=int32>]
cube_func_int32.graph.get_operation_by_name("x")
輸出:
<tf.Operation 'x' type=Placeholder>
cube_func_int32.graph.get_tensor_by_name("x:0") #默認(rèn)加“:0”
<tf.Tensor 'x:0' shape=(None,) dtype=int32>
cube_func_int32.graph.as_graph_def() #總名字,針對上面兩個(gè)
node { name: "x" op: "Placeholder" attr { key: "_user_specified_name" value { s: "x" } } attr { key: "dtype" value { type: DT_INT32 } } attr { key: "shape" value { shape { dim { size: -1 } } } } } node { name: "Pow/y" op: "Const" attr { key: "dtype" value { type: DT_INT32 } } attr { key: "value" value { tensor { dtype: DT_INT32 tensor_shape { } int_val: 3 } } } } node { name: "Pow" op: "Pow" input: "x" input: "Pow/y" attr { key: "T" value { type: DT_INT32 } } } node { name: "Identity" op: "Identity" input: "Pow" attr { key: "T" value { type: DT_INT32 } } } versions { producer: 119 }
到此這篇關(guān)于tensorflow2.0的函數(shù)簽名與圖結(jié)構(gòu)的文章就介紹到這了,更多相關(guān)tensorflow函數(shù)簽名與圖結(jié)構(gòu)內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!
相關(guān)文章
運(yùn)行獨(dú)立 pyspark 時(shí)出現(xiàn) Windows 錯(cuò)誤解決辦法
在本篇文章里小編給大家分享的是一篇關(guān)于運(yùn)行獨(dú)立 pyspark 時(shí)出現(xiàn) Windows 錯(cuò)誤解決辦法,對此有需求的方法可以參考下。2021-12-12python中matplotlib實(shí)現(xiàn)最小二乘法擬合的過程詳解
這篇文章主要給大家介紹了關(guān)于python中matplotlib實(shí)現(xiàn)最小二乘法擬合的相關(guān)資料,文中通過示例代碼詳細(xì)介紹了關(guān)于最小二乘法擬合直線和最小二乘法擬合曲線的實(shí)現(xiàn)過程,需要的朋友可以參考借鑒,下面來一起看看吧。2017-07-07Pycharm中切換pytorch的環(huán)境和配置的教程詳解
這篇文章主要介紹了Pycharm中切換pytorch的環(huán)境和配置,本文給大家介紹的非常詳細(xì),對大家的工作或?qū)W習(xí)具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2020-03-03Pandas_cum累積計(jì)算和rolling滾動(dòng)計(jì)算的用法詳解
今天小編就為大家分享一篇Pandas_cum累積計(jì)算和rolling滾動(dòng)計(jì)算的用法詳解,具有好的參考價(jià)值,希望對大家有所幫助。一起跟隨小編過來看看吧2019-07-07解決python升級引起的pip執(zhí)行錯(cuò)誤的問題
今天小編就為大家分享一篇解決python升級引起的pip執(zhí)行錯(cuò)誤的問題,具有很好的參考價(jià)值,希望對大家有所幫助。一起跟隨小編過來看看吧2018-06-06Python爬蟲小技巧之偽造隨機(jī)的User-Agent
這篇文章主要給大家介紹了關(guān)于Python爬蟲小技巧之偽造隨機(jī)的User-Agent的相關(guān)資料,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2018-09-09Django中和時(shí)區(qū)相關(guān)的安全問題詳解
這篇文章主要給大家介紹了關(guān)于Django中和時(shí)區(qū)相關(guān)的安全問題的相關(guān)資料,需要的朋友可以參考下2020-10-10