java 操作 TensorFlow 合适吗?

Question

java 操作 TensorFlow 合适吗

Answer 1

tensorflow训练模型通常使用Python api编写，这些模型保存后可以在Java中调用。python中训练完成，模型保存使用如下api保存：保存为二进制pb文件，主要的点是output_node_names数组，该数据的名称表示需要保存的tensorflow tensor名。既是在python中定义模型时指定的计算操作的name。填写什么就保存到什么节点。在cnn模型中，通常是分类输出的名称。例如模型定义时代码为：模型在java中使用需要关心模型输入tensor和输出tensor名，所以定义模型时，所有的输入tensor最好指定名称，如输入x和dropout名。java中调用代码片段：