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 相关文章推荐
在Django的URLconf中使用命名组的方法
Jul 18 Python
python基础入门学习笔记(Python环境搭建)
Jan 13 Python
Python基于matplotlib绘制栈式直方图的方法示例
Aug 09 Python
Python使用wxPython实现计算器
Jan 30 Python
Python实现判断并移除列表指定位置元素的方法
Apr 13 Python
使用Python通过win32 COM实现Word文档的写入与保存方法
May 08 Python
详解Django+Uwsgi+Nginx的生产环境部署
Jun 25 Python
局域网内python socket实现windows与linux间的消息传送
Apr 19 Python
在PyCharm的 Terminal(终端)切换Python版本的方法
Aug 02 Python
python集合的创建、添加及删除操作示例
Oct 08 Python
如何基于Python代码实现高精度免费OCR工具
Jun 18 Python
Django如何创作一个简单的最小程序
May 12 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中用PHPMailer来发送邮件的示例(126.com的例子)
2013/06/24 PHP
php指定函数参数默认值示例代码
2013/12/04 PHP
PHP下的浮点运算不准的解决方法
2016/10/27 PHP
简单谈谈 php 文件锁
2017/02/19 PHP
php读取出一个文件夹及其子文件夹下所有文件的方法示例
2017/06/15 PHP
PHP实现的迪科斯彻(Dijkstra)最短路径算法实例
2017/09/16 PHP
PHP文件管理之实现网盘及压缩包的功能操作
2017/09/20 PHP
PHP从零开始打造自己的MVC框架之入口文件实现方法详解
2019/06/03 PHP
jQuery 使用手册(五)
2009/09/23 Javascript
浅谈javascript面向对象程序设计
2015/01/21 Javascript
jQuery监控文本框事件并作相应处理的方法
2015/04/16 Javascript
在AngularJS框架中处理数据建模的方式解析
2016/03/05 Javascript
BootStrap创建响应式导航条实例代码
2016/05/31 Javascript
又一枚精彩的弹幕效果jQuery实现
2016/07/25 Javascript
理解AngularJs篇:30分钟快速掌握AngularJs
2016/12/23 Javascript
为什么我们要做三份 Webpack 配置文件
2017/09/18 Javascript
Vue基于NUXT的SSR详解
2017/10/24 Javascript
JavaScript中set与get方法用法示例
2018/08/15 Javascript
element-ui中el-upload多文件一次性上传的实现
2020/12/02 Javascript
[00:10]神之谴戒
2019/03/06 DOTA
详谈Python基础之内置函数和递归
2017/06/21 Python
基于Python列表解析(列表推导式)
2018/06/23 Python
python解析xml简单示例
2019/06/21 Python
termux中matplotlib无法显示中文问题的解决方法
2021/01/11 Python
涂鸦板简单实现 Html5编写属于自己的画画板
2016/07/05 HTML / CSS
日本面向世界,国际级的免税在线购物商城:DOKODEMO
2017/02/01 全球购物
俄罗斯金苹果网上化妆品和香水商店:Goldapple
2019/12/01 全球购物
包装类的功能、种类、常用方法
2012/01/27 面试题
会计应届生的自荐信
2013/12/13 职场文书
大学生学年自我鉴定
2014/02/10 职场文书
环保建议书500字
2014/05/14 职场文书
大学生党员自我批评思想汇报
2014/10/10 职场文书
交警正风肃纪剖析材料
2014/10/29 职场文书
2014年维稳工作总结
2014/11/18 职场文书
2014年医院科室工作总结
2014/12/20 职场文书
Golang连接并操作MySQL
2022/04/14 MySQL