1. 程式人生 > >【Tensorflow】tf.import_graph_def

【Tensorflow】tf.import_graph_def

tf.import_graph_def

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

將圖從graph_def匯入到當前預設圖中. (即將捨棄的引數)

SOME ARGUMENTS ARE DEPRECATED. 它們將在未來的版本中被刪除。更新說明:如果你需要此特性請在 https://github.com/tensorflow/tensorflow/issues 提出。

這個函式提供了一種方法來匯入序列化的TensorFlow GraphDef協議緩衝區,並將GraphDef中的各個物件提取為tf。張量和tf。操作物件。一旦提取出來,這些物件就會被放到當前的預設圖形中。。參考tf.Graph.as_graph_def 來建立 a GraphDef proto.

引數:

  • graph_def: 包含要匯入到預設圖中的操作的GraphDef proto。
  • input_map: 將graph_def中的輸入名稱(作為字串)對映到張量物件的字典。輸入圖中指定的輸入張量的值將被重新對映到相應的張量值。
  • return_elements: 在graph_def中包含操作名的字串列表,將作為operationobject返回;和/或graph_def中的張量名稱,它們將作為張量物件返回。
  • name: (可選.) 將字首放在graph_def中名稱前面的字首。注意,這並不適用於匯入的函式名。預設為"import".
  • op_dict: (可選.) 已棄用,請勿使用
  • producer_op_list: (可選.) 一個OpList原型,帶有(可能是剝離的)圖表生產者使用的OpDefs列表。如果提供了,那麼根據producer_op_list的預設值,在graph_def中無法識別的ops attrs將被刪除。這將允許稍後的二進位制檔案生成更多的graphdef被早期的二進位制檔案所接受。

返回:

   從匯入的圖中得到的與return_element中的名稱相對應的操作和/或張量物件的列表。

錯誤型別:

  • TypeError: 如果graph_def不是GraphDef proto, input_map不是字典對映字串到Tensorobjects,或者return_elements不是字串列表。
  • ValueError: 如果input_map,或者return_elements包含在graph_def中沒有出現的名稱,或者graph_def格式不好(例如,它指的是一個未知張量)。