tensorflow对pd文件的保存和加载

最新推荐文章于 2021-06-22 23:28:07 发布

原创

最新推荐文章于 2021-06-22 23:28:07 发布 · 1.3k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#tensorflow

本文介绍了使用TensorFlow创建简单模型的过程，包括变量定义、占位符使用、运算操作及模型的保存与重新加载。通过具体代码示例，展示了如何将训练后的模型转化为.pb文件，并在新的会话中读取模型进行预测。

代码只是一份笔记，以免以后忘记了

import tensorflow as tf

from tensorflow.python.framework import graph_util

var1 = tf.Variable(1.0, dtype=tf.float32, name='v1')

var2 = tf.Variable(2.0, dtype=tf.float32, name='v2')

var3 = tf.Variable(2.0, dtype=tf.float32, name='v3')

x = tf.placeholder(dtype=tf.float32, shape=None