TensorFlow圖形處理函數(shù)：tf.import_graph_def

函數(shù)：tf.import_graph_def

import_graph_def(
    graph_def,
    input_map=None,
    return_elements=None,
    name=None,
    op_dict=None,
    producer_op_list=None
)

定義在：tensorflow/python/framework/importer.py.

參見指南：構(gòu)建圖>實用功能

將圖形從 graph_def 導(dǎo)入當(dāng)前的默認(rèn) Graph.

此函數(shù)提供了一種導(dǎo)入序列化的 TensorFlow GraphDef 協(xié)議緩沖區(qū)的方法,并將 GraphDefas 中的各個對象提取為 tf.Tensor 和 tf.Operation 對象.一旦提取,這些對象將被放置到當(dāng)前的默認(rèn) Graph 中.參考 tf.Graph.as_graph_def 創(chuàng)建一個 GraphDef 原型.

參數(shù)：

• graph_def：一個 GraphDef 原型,它包含了要導(dǎo)入默認(rèn)圖形的操作.
• input_map：將 graph_def 中的輸入名稱 (作為字符串) 映射到張量對象的字典.導(dǎo)入的圖形中的指定輸入張量的值將 re-mapped 到各自的幅值.
• return_elements：包含將作為操作對象返回的 graph_def 中的操作名稱的字符串列表或張量名稱在 graph_def,將作為張量對象返回.
• name：(可選)將在 graph_def 中預(yù)先處理的前綴.請注意,這不適用于導(dǎo)入的函數(shù)名稱.默認(rèn)為 "import".
  
• op_dict：(可選)將 op 類型名稱映射到 OpDefprotos 的字典.必須包含在 graph_def 中命名的每個 op 類型的 OpDef 原型.如果省略,請使用在全局注冊表中注冊的 OpDef 原型.
  
• producer_op_list：(可選)由圖的生產(chǎn)者使用的 (可能被剝掉的) OpDefs 的 OpList 原始列表.如果提供,attrs 的 ops 在 graph_def 而不在 op_dict,根據(jù)producer_op_list 的默認(rèn)值 將被刪除.這將允許后面的二進(jìn)制文件所產(chǎn)生的一些 GraphDefs 被早期的二進(jìn)制文件接受.

返回值：

該函數(shù)返回來自導(dǎo)入圖形的操作或張量對象的列表,對應(yīng)于 return_elements 中的名稱

可能引發(fā)的異常：

• TypeError：如果 graph_def 不是 GraphDef 原型,input_map 不是一個將字符串映射到張量對象的字典,或者 return_elements 不是字符串列表.
• ValueError：如果 input_map,或 return_elements 包含的名稱不出現(xiàn)在 graph_def,或 graph_def 是不正確的 (例如,它是指一個未知的張量).