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.

參見(jiàn)指南：構(gòu)建圖>實(shí)用功能

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

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

參數(shù)：

• graph_def：一個(gè) GraphDef 原型,它包含了要導(dǎo)入默認(rèn)圖形的操作.
• input_map：將 graph_def 中的輸入名稱(chēng) (作為字符串) 映射到張量對(duì)象的字典.導(dǎo)入的圖形中的指定輸入張量的值將 re-mapped 到各自的幅值.
• return_elements：包含將作為操作對(duì)象返回的 graph_def 中的操作名稱(chēng)的字符串列表或張量名稱(chēng)在 graph_def,將作為張量對(duì)象返回.
• name：(可選)將在 graph_def 中預(yù)先處理的前綴.請(qǐng)注意,這不適用于導(dǎo)入的函數(shù)名稱(chēng).默認(rèn)為 "import".
  
• op_dict：(可選)將 op 類(lèi)型名稱(chēng)映射到 OpDefprotos 的字典.必須包含在 graph_def 中命名的每個(gè) op 類(lèi)型的 OpDef 原型.如果省略,請(qǐng)使用在全局注冊(cè)表中注冊(cè)的 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ù)返回來(lái)自導(dǎo)入圖形的操作或張量對(duì)象的列表,對(duì)應(yīng)于 return_elements 中的名稱(chēng)

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

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