tensorflow获取变量维度信息
作者:shuzfan 时间:2021-01-30 01:21:41
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]
来源:http://blog.csdn.net/shuzfan/article/details/79051042
标签:tensorflow,变量
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
Django 多环境配置详解
2022-01-25 06:38:49
ASP教程:自己写的数据库操作类
2008-11-21 17:29:00
精心整理总结的Python自动化测试面试题
2022-07-27 08:13:00
![](https://img.aspxhome.com/file/2023/4/91934_0s.png)
Python实现字符串中某个字母的替代功能
2021-09-28 13:13:02
tensorflow基于CNN实战mnist手写识别(小白必看)
2021-01-26 22:33:42
![](https://img.aspxhome.com/file/2023/2/99432_0s.png)
关于python中密码加盐的学习体会小结
2022-05-13 09:47:04
Python量化交易实战之使用Resample函数转换“日K”数据
2023-07-30 07:31:11
![](https://img.aspxhome.com/file/2023/2/105232_0s.png)
django celery redis使用具体实践
2022-12-03 02:45:39
![](https://img.aspxhome.com/file/2023/5/87785_0s.png)
使用Python文件读写,自定义分隔符(custom delimiter)
2021-12-06 08:17:51
asp中Adodb.Stream 的使用说明
2007-09-11 13:53:00
Sun正式发布MySQL 5.1版 简化数据库应用
2008-12-11 15:15:00
Pycharm直接使用远程服务器代码并调试的解决方法
2021-05-13 10:21:03
Python实现朴素贝叶斯的学习与分类过程解析
2023-08-17 15:15:23
![](https://img.aspxhome.com/file/2023/2/67742_0s.jpg)
django基于存储在前端的token用户认证解析
2023-12-27 19:14:59
![](https://img.aspxhome.com/file/2023/5/103485_0s.png)
关于python中plt.hist参数的使用详解
2021-07-05 05:57:23
![](https://img.aspxhome.com/file/2023/9/88909_0s.jpg)
javascript闭包的秘密
2008-09-28 20:39:00
Jquery实现div模拟Select控件
2008-12-01 15:47:00
Microsoft Access项目不能压缩的原因
2008-11-28 14:48:00
js正则的几个基本概念
2007-09-30 20:02:00
python中执行shell的两种方法总结
2023-03-12 15:20:39