今天小编就为大家分享一篇tensorflow 保存模型和取出中间权重例子，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

下面代码的功能是先训练一个简单的模型，然后保存模型，同时保存到一个pb文件当中，后续可以从pd文件里读取权重值。 import tensorflow as tf import numpy as np import os import h5py import pickle from tensorflow.python.framework import graph_util from tensorflow.python.platform import gfile #设置使用指定GPU os.env