Tensorflow获取张量Tensor的具体维数实例
作者:sigma_Tian 时间:2021-12-24 20:25:10
获取Tensor的维数
>>> import tensorflow as tf
>>> tf.__version__
'1.2.0-rc1'
>>> x=tf.placeholder(dtype=float32,shape=[1,2,3,4])
>>> x=tf.placeholder(dtype=tf.float32,shape=[1,2,3,4])
>>> x.shape
TensorShape([Dimension(1), Dimension(2), Dimension(3), Dimension(4)])
>>> x.get_shape()
TensorShape([Dimension(1), Dimension(2), Dimension(3), Dimension(4)])
# 返回tuple
>>> x.shape[2]
Dimension(3)
>>> x.get_shape()[2]
Dimension(3)
# 获取具体维度数值
>>> x.shape[2].value
3
>>> x.get_shape()[2].value
3
# 也可以将TensorShape变量转化为list类型,然后直接按照索引取值
>>> x.shape.as_list()
[1, 2, 3, 4]
>>> x.shape.as_list()
[1, 2, 3, 4]
# 可以与int型数值比较
>>> x.shape[2] == 3
True
>>> x.get_shape()[2] == 3
True
来源:https://blog.csdn.net/tcx1992/article/details/80249992
标签:Tensorflow,张量,Tensor,维数
0
投稿
猜你喜欢
python判断文件是否存在,不存在就创建一个的实例
2022-04-29 02:28:55
PHP三层结构(上) 简单三层结构
2023-06-20 06:36:07
Python 动态变量名定义与调用方法
2023-07-29 22:36:05
SQL进行排序、分组、统计的10个新技巧
2009-01-23 13:59:00
Python中for后接else的语法使用
2023-07-08 15:28:47
ASPJPEG组件使用详解(缩略图+水印)
2007-09-19 17:31:00
浅谈django开发者模式中的autoreload是如何实现的
2022-04-03 17:22:04
Pycharm及python安装详细教程(图解)
2023-01-06 00:26:00
基于Python log 的正确打开方式
2021-05-29 21:42:59
sqlserver 数据库连接字符串中的可选项收集
2011-10-24 19:48:37
初学ASP编程易犯的一个错误要注意
2008-11-07 15:08:00
caffe binaryproto 与 npy相互转换的实例讲解
2021-10-22 15:38:48
Python蛇形方阵的实现
2023-06-27 16:33:21
DropDownList绑定选择数据报错提示异常解决方案
2023-07-18 04:36:13
详解Python中表达式i += x与i = i + x是否等价
2023-07-29 06:51:12
Go程序性能优化及pprof使用方法详解
2023-08-28 14:04:40
Python+OpenCV图像处理——实现轮廓发现
2023-12-12 23:30:31
单选的radia,和多选的checkbox如何判定选择?
2008-11-26 12:29:00
详解pycharm配置python解释器的问题
2022-10-14 13:22:08
ASP基础教程之学习ASP中子程序的应用
2008-10-16 10:53:00