tensorflow获取变量维度信息


Posted in Python onMarch 10, 2018

tensorflow版本1.4

获取变量维度是一个使用频繁的操作,在tensorflow中获取变量维度主要用到的操作有以下三种:

  • Tensor.shape
  • Tensor.get_shape()
  • tf.shape(input,name=None,out_type=tf.int32)

对上面三种操作做一下简单分析:(这三种操作先记作A、B、C)

A 和 B 基本一样,只不过前者是Tensor的属性变量,后者是Tensor的函数。
A 和 B 均返回TensorShape类型,而 C 返回一个1D的out_type类型的Tensor。
A 和 B 可以在任意位置使用,而 C 必须在Session中使用。
A 和 B 获取的是静态shape,可以返回不完整的shape; C 获取的是动态的shape,必须是完整的shape。

另外,补充从TenaorShape变量中获取具体维度数值的方法

# 直接获取TensorShape变量的第i个维度值
x.shape[i].value
x.get_shape()[i].value

# 将TensorShape变量转化为list类型,然后直接按照索引取值
x.get_shape().as_list()

下面给出全部的示例程序:

import tensorflow as tf

x1 = tf.constant([[1,2,3],[4,5,6]])
# 占位符创建变量,第一个维度初始化为None,表示暂不指定维度
x2 = tf.placeholder(tf.float32,[None, 2,3])
print('x1.shape:',x1.shape)
print('x2.shape:',x2.shape)
print('x2.shape[1].value:',x2.shape[1].value)
print('tf.shape(x1):',tf.shape(x1))
print('tf.shape(x2):',tf.shape(x2))
print('x1.get_shape():',x1.get_shape())
print('x2.get_shape():',x2.get_shape())
print('x2.get_shape.as_list[1]:',x2.get_shape().as_list()[1])
shapeOP1 = tf.shape(x1)
shapeOP2 = tf.shape(x2)
with tf.Session() as sess:
 print('Within session, tf.shape(x1):',sess.run(shapeOP1))
 # 由于x2未进行完整的变量填充,其维度不完整,因此执行下面的命令将会报错
 # print('Within session, tf.shape(x2):',sess.run(shapeOP2)) # 此命令将会报错

输出结果为:

x1.shape: (2, 3)
x2.shape: (?, 2, 3)
x2.shape[1].value: 2
tf.shape(x1): Tensor("Shape:0", shape=(2,), dtype=int32)
tf.shape(x2): Tensor("Shape_1:0", shape=(3,), dtype=int32)
x1.get_shape(): (2, 3)
x2.get_shape(): (?, 2, 3)
x2.get_shape.as_list[1]: 2
Within session, tf.shape(x1): [2 3]

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持三水点靠木。

Python 相关文章推荐
部署Python的框架下的web app的详细教程
Apr 30 Python
Python使用Scrapy爬取妹子图
May 28 Python
python类中super()和__init__()的区别
Oct 18 Python
Python之re操作方法(详解)
Jun 14 Python
Python实现修改文件内容的方法分析
Mar 25 Python
Python上下文管理器类和上下文管理器装饰器contextmanager用法实例分析
Nov 07 Python
解决pandas展示数据输出时列名不能对齐的问题
Nov 18 Python
pytorch实现mnist数据集的图像可视化及保存
Jan 14 Python
Django使用list对单个或者多个字段求values值实例
Mar 31 Python
Python Selenium破解滑块验证码最新版(GEETEST95%以上通过率)
Jan 29 Python
再也不用花钱买漫画!Python爬取某漫画的脚本及源码
Jun 09 Python
Python Matplotlib绘制动画的代码详解
May 30 Python
TensorFlow变量管理详解
Mar 10 #Python
TensorFlow神经网络优化策略学习
Mar 09 #Python
TensorFlow实现AutoEncoder自编码器
Mar 09 #Python
TensorFlow实现MLP多层感知机模型
Mar 09 #Python
TensorFlow实现Softmax回归模型
Mar 09 #Python
用python实现百度翻译的示例代码
Mar 09 #Python
TensorFlow深度学习之卷积神经网络CNN
Mar 09 #Python
You might like
腾讯QQ微博API接口获取微博内容
2013/10/30 PHP
Linux php 中文乱码的快速解决方法
2016/05/13 PHP
TP5框架简单登录功能实现方法示例
2019/10/31 PHP
php的lavarel框架中join和orWhere的用法
2020/12/28 PHP
Javascript实现的分页函数
2006/12/22 Javascript
javascript脚本编程解决考试分数统计问题
2008/10/18 Javascript
js 鼠标点击事件及其它捕获
2009/06/04 Javascript
精通Javascript系列之数据类型 字符串
2011/06/08 Javascript
js实现选中复选框文字变色的方法
2015/08/14 Javascript
angularjs表格ng-table使用备忘录
2016/03/09 Javascript
老生常谈js动态添加事件--- 事件委托
2016/07/19 Javascript
浅谈javascript alert和confirm的美化
2016/12/15 Javascript
bootstrap为水平排列的表单和内联表单设置可选的图标
2017/02/15 Javascript
详解用vue-cli来搭建vue项目和webpack
2017/04/20 Javascript
自定义PC微信扫码登录样式写法
2017/12/12 Javascript
详解webpack3编译兼容IE8的正确姿势
2017/12/21 Javascript
JS中实现隐藏部分姓名或者电话号码的代码
2018/07/17 Javascript
layui-laydate时间日历控件使用方法详解
2018/11/15 Javascript
JS实现盒子跟着鼠标移动及键盘方向键控制盒子移动效果示例
2019/01/29 Javascript
基于vue-cli搭建多模块且各模块独立打包的项目
2019/06/12 Javascript
详解利用nodejs对本地json文件进行增删改查
2019/09/20 NodeJs
JS几个常用的函数和对象定义与用法示例
2020/01/15 Javascript
vue监听dom大小改变案例
2020/07/29 Javascript
python实现在pickling的时候压缩的方法
2014/09/25 Python
Tensorflow 查看变量的值方法
2018/06/14 Python
利用rest framework搭建Django API过程解析
2019/08/31 Python
Python 将json序列化后的字符串转换成字典(推荐)
2020/01/06 Python
python入门之基础语法学习笔记
2020/02/08 Python
详解如何修改python中字典的键和值
2020/09/29 Python
CSS3实现莲花绽放的动画效果
2020/11/06 HTML / CSS
美国折衷生活方式品牌:Robert Graham
2018/07/13 全球购物
Keds加拿大官网:购买帆布运动鞋和皮鞋
2019/09/26 全球购物
Java平台和其他软件平台有什么不同
2015/06/05 面试题
心理健康教育制度
2014/01/27 职场文书
思想工作总结范文
2015/08/12 职场文书
详解Python中的进程和线程
2021/06/23 Python