tensorflow:加载预训练网络及获取operation、tensor值

最新推荐文章于 2021-08-17 09:52:55 发布

dongtuu

最新推荐文章于 2021-08-17 09:52:55 发布

阅读量2.4k

点赞数 1

分类专栏： tensorflow 文章标签： tensorflow

本文链接：https://blog.csdn.net/h2oco2ch4/article/details/80903366

版权

本文是关于TensorFlow中加载预训练网络以及获取tensor和operation的方法的个人笔记。介绍了通过不同方式获取计算图中的变量，并指出计算图中的tensor和operation通常是相伴出现的。文中提供两种获取tensor的方法，但强调没有明确的名字时查找的困难性。

摘要由CSDN通过智能技术生成

下文只是个人笔记，不保证正确和详尽。

tensorflow的计算图包括tensor、operation。保存一张完整的计算图意味着同时保存两者，只需要一句话。但加载网络有痛处，在于无法当你不是作者的时候，无法快速得知网络里到底有哪些tensor、operation。

计算图里的tensor和operation几乎结伴出现（我没验证过）。定义一个variable，则它也配备一个operation；设计一个operation，则它也有对应的输出作为tensor。所有调用格式都是：

。	。	eg
tensor	name:index	w:0
operation	name	w

index怎么取的还没搞懂。
如果没有名字（这取决于一开始训练有没有取名字），那么只能用其他接口查阅有哪些名字，再调用，下文会提。

这是我测试用的网络。注释掉的部分可以用来做对比。
tf早期的图、权重保存方式和现在的不一样，现在是分3个文件保存的：.data .meta .index，另外还有一个记录文件checkpoint。详情谷歌。

def first():
    W = tf.Variable([[1,1,1],[2,2,2]],dtype = tf.float32,name=<

关注

专栏目录