tensorflow訓(xùn)練模型通常使用Python api編寫，這些模型保存后可以在Java中調(diào)用。

python中訓(xùn)練完成，模型保存使用如下api保存：

保存為二進(jìn)制pb文件，主要的點(diǎn)是output_node_names數(shù)組，該數(shù)據(jù)的名稱表示需要保存的tensorflow tensor名。既是在python中定義模型時指定的計算操作的name。填寫什么就保存到什么節(jié)點(diǎn)。在cnn模型中，通常是分類輸出的名稱。

例如模型定義時代碼為：

模型在java中使用需要關(guān)心模型輸入tensor和輸出tensor名，所以定義模型時，所有的輸入tensor最好指定名稱，如輸入x和dropout名。

java中調(diào)用代碼片段：