【Tensorflow】tf.import_graph_def
阿新 • • 發佈:2018-12-09
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格式不好(例如,它指的是一個未知張量)。