TensorFlow查看输入节点和输出节点名称方式


Posted in Python onJanuary 04, 2020

TensorFlow 定义输入节点名称input_name:

with tf.name_scope('input'):
  bottleneck_input = tf.placeholder_with_default(
    bottleneck_tensor,
    shape=[batch_size, bottleneck_tensor_size],
    name='Mul')

TensorFlow查看pb数据库里面的输入节点和输出节点:

import tensorflow as tf
import os
 
model_dir = './tmp/'
model_name = 'output_graph.pb'
 
def create_graph():
  with tf.gfile.FastGFile(os.path.join(
      model_dir, model_name), 'rb') as f:
    graph_def = tf.GraphDef()
    graph_def.ParseFromString(f.read())
    tf.import_graph_def(graph_def, name='')
 
create_graph()
tensor_name_list = [tensor.name for tensor in tf.get_default_graph().as_graph_def().node]
for tensor_name in tensor_name_list:
  print(tensor_name,'\n')

以上这篇TensorFlow查看输入节点和输出节点名称方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
简单讲解Python中的字符串与字符串的输入输出
Mar 13 Python
python 队列详解及实例代码
Oct 18 Python
用python与文件进行交互的方法
Mar 01 Python
python获取文件真实链接的方法,针对于302返回码
May 14 Python
python实现彩票系统
Jun 28 Python
Python实现多级目录压缩与解压文件的方法
Sep 01 Python
Python判断是否json是否包含一个key的方法
Dec 31 Python
python中使用 xlwt 操作excel的常见方法与问题
Jan 13 Python
windows上安装python3教程以及环境变量配置详解
Jul 18 Python
Python参数传递机制传值和传引用原理详解
May 22 Python
Python爬虫之Spider类用法简单介绍
Aug 04 Python
Django模型层实现多表关系创建和多表操作
Jul 21 Python
根据tensor的名字获取变量的值方式
Jan 04 #Python
将tensorflow.Variable中的某些元素取出组成一个新的矩阵示例
Jan 04 #Python
tensorflow实现tensor中满足某一条件的数值取出组成新的tensor
Jan 04 #Python
对tensorflow中的strides参数使用详解
Jan 04 #Python
tensorflow之获取tensor的shape作为max_pool的ksize实例
Jan 04 #Python
TensorFlow tf.nn.max_pool实现池化操作方式
Jan 04 #Python
TensorFlow tf.nn.conv2d实现卷积的方式
Jan 03 #Python
You might like
一篇入门的php Class 文章
2007/04/04 PHP
mysql 中InnoDB和MyISAM的区别分析小结
2008/04/15 PHP
thinkphp视图模型查询提示ERR: 1146:Table 'db.pr_order_view' doesn't exist的解决方法
2014/10/30 PHP
PHP程序中使用adodb连接不同数据库的代码实例
2015/12/19 PHP
php解决DOM乱码的方法示例代码
2016/11/20 PHP
php写一个函数,实现扫描并打印出自定目录下(含子目录)所有jpg文件名
2017/05/26 PHP
PHP堆栈调试操作简单示例
2018/06/15 PHP
PHP number_format函数原理及实例解析
2020/07/14 PHP
setTimeout与setInterval在不同浏览器下的差异
2010/01/24 Javascript
javascript时区函数介绍
2012/09/14 Javascript
js怎么终止程序return不行换jfslk
2013/05/30 Javascript
js jquery ajax的几种用法总结(及优缺点介绍)
2014/01/28 Javascript
JS的数组迭代方法
2015/02/05 Javascript
jQuery实现网页顶部固定导航效果代码
2015/12/24 Javascript
javascript拖拽应用实例
2016/03/25 Javascript
基于JS实现密码框(password)中显示文字提示功能代码
2016/05/27 Javascript
一句jQuery代码实现返回顶部效果(简单实用)
2016/12/28 Javascript
javascript工厂模式和构造函数模式创建对象方法解析
2016/12/30 Javascript
详解JavaScript 新语法之Class 的私有属性与私有方法
2019/04/23 Javascript
javascript防抖函数debounce详解
2019/06/11 Javascript
vue递归组件实战之简单树形控件实例代码
2019/08/27 Javascript
如何在JavaScript中等分数组的实现
2020/12/13 Javascript
Python中dictionary items()系列函数的用法实例
2014/08/21 Python
Python实现的石头剪子布代码分享
2014/08/22 Python
python查询mysql,返回json的实例
2018/03/26 Python
如何利用Python识别图片中的文字
2020/05/31 Python
Python  Asyncio模块实现的生产消费者模型的方法
2021/03/01 Python
澳大利亚时尚前卫设计师珠宝在线:Amber Sceats
2017/10/04 全球购物
技术岗位竞聘演讲稿
2014/05/16 职场文书
企业文化标语口号
2014/06/09 职场文书
结婚保证书
2015/01/16 职场文书
党员干部廉洁自律承诺书
2015/04/28 职场文书
2015年青年志愿者工作总结
2015/05/20 职场文书
银行求职信怎么写
2019/06/20 职场文书
MySQL 全文索引使用指南
2021/05/25 MySQL
python+pyhyper实现识别图片中的车牌号思路详解
2022/12/24 Python