TensorFlow用expand_dim()来增加维度的方法


Posted in Python onJuly 26, 2018

TensorFlow中,想要维度增加一维,可以使用tf.expand_dims(input, dim, name=None)函数。当然,我们常用tf.reshape(input, shape=[])也可以达到相同效果,但是有些时候在构建图的过程中,placeholder没有被feed具体的值,这时就会包下面的错误:TypeError: Expected binary or unicode string, got 1

在这种情况下,我们就可以考虑使用expand_dims来将维度加1。比如我自己代码中遇到的情况,在对图像维度降到二维做特定操作后,要还原成四维[batch, height, width, channels],前后各增加一维。如果用reshape,则因为上述原因报错

one_img2 = tf.reshape(one_img, shape=[1, one_img.get_shape()[0].value, one_img.get_shape()[1].value, 1])

用下面的方法可以实现:

one_img = tf.expand_dims(one_img, 0)
one_img = tf.expand_dims(one_img, -1) #-1表示最后一维

在最后,给出官方的例子和说明

# 't' is a tensor of shape [2]
shape(expand_dims(t, 0)) ==> [1, 2]
shape(expand_dims(t, 1)) ==> [2, 1]
shape(expand_dims(t, -1)) ==> [2, 1]

# 't2' is a tensor of shape [2, 3, 5]
shape(expand_dims(t2, 0)) ==> [1, 2, 3, 5]
shape(expand_dims(t2, 2)) ==> [2, 3, 1, 5]
shape(expand_dims(t2, 3)) ==> [2, 3, 5, 1]

Args:

input: A Tensor.
dim: A Tensor. Must be one of the following types: int32, int64. 0-D (scalar). Specifies the dimension index at which to expand the shape of input.
name: A name for the operation (optional).

Returns:

A Tensor. Has the same type as input. Contains the same data as input, but its shape has an additional dimension of size 1 added.

以上这篇TensorFlow用expand_dim()来增加维度的方法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
linux系统使用python监控apache服务器进程脚本分享
Jan 15 Python
python 调用HBase的简单实例
Dec 18 Python
Flask框架使用DBUtils模块连接数据库操作示例
Jul 20 Python
python中scikit-learn机器代码实例
Aug 05 Python
Python爬虫 scrapy框架爬取某招聘网存入mongodb解析
Jul 31 Python
ipad上运行python的方法步骤
Oct 12 Python
PyTorch实现AlexNet示例
Jan 14 Python
python爬虫请求头设置代码
Jul 28 Python
python实现三种随机请求头方式
Jan 05 Python
python将图片转为矢量图的方法步骤
Mar 30 Python
一劳永逸彻底解决pip install慢的办法
May 24 Python
如何解决.cuda()加载用时很长的问题
May 24 Python
Python迭代器与生成器基本用法分析
Jul 26 #Python
Python实现矩阵相乘的三种方法小结
Jul 26 #Python
python实现泊松图像融合
Jul 26 #Python
python中的decorator的作用详解
Jul 26 #Python
python opencv实现旋转矩形框裁减功能
Jul 25 #Python
Python3匿名函数用法示例
Jul 25 #Python
Python实现动态添加属性和方法操作示例
Jul 25 #Python
You might like
用mysql_fetch_array()获取当前行数据的方法详解
2013/06/05 PHP
php通过exif_read_data函数获取图片的exif信息
2015/05/21 PHP
php+redis实现多台服务器内网存储session并读取示例
2017/01/12 PHP
通过JavaScript控制字体大小的代码
2011/10/04 Javascript
JS实现点击下载的小例子
2013/07/10 Javascript
getJSON调用后台json数据时函数被调用两次的原因猜想
2013/09/29 Javascript
深入理解Javascript动态方法调用与参数修改的问题
2013/12/10 Javascript
js身份证判断方法支持15位和18位
2014/03/18 Javascript
jquery的幻灯片图片切换效果代码分享
2015/09/07 Javascript
jQuery操作Table技巧大汇总
2016/01/23 Javascript
javascript中eval解析JSON字符串
2016/02/27 Javascript
Javascript 5种方法实现过滤删除前后所有空格
2016/06/22 Javascript
微信小程序(应用号)开发新闻客户端实例
2016/10/24 Javascript
JS根据生日月份和日期计算星座的简单实现方法
2016/11/24 Javascript
Nodejs调用WebService的示例代码
2017/09/29 NodeJs
javascript 中模板方法单例的实现方法
2017/10/17 Javascript
vue2.0 循环遍历加载不同图片的方法
2018/03/06 Javascript
vue项目如何刷新当前页面的方法
2018/05/18 Javascript
vue v-model动态生成详解
2018/06/30 Javascript
vuex实现购物车的增加减少移除
2020/06/28 Javascript
[03:39]2015国际邀请赛主赛事首日精彩回顾
2015/08/05 DOTA
Python ZipFile模块详解
2013/11/01 Python
Python实现多线程下载文件的代码实例
2014/06/01 Python
利用python求相邻数的方法示例
2017/08/18 Python
Python3.6通过自带的urllib通过get或post方法请求url的实例
2018/05/10 Python
python爬虫之模拟登陆csdn的实例代码
2018/05/18 Python
python实现批量nii文件转换为png图像
2019/07/18 Python
通过Python扫描代码关键字并进行预警的实现方法
2020/05/24 Python
python获取系统内存占用信息的实例方法
2020/07/17 Python
html5 canvas实现跟随鼠标旋转的箭头
2016/03/11 HTML / CSS
计算机软件个人的自荐信范文
2013/12/01 职场文书
旅游管理专业个人求职信范文
2013/12/24 职场文书
平安工地建设方案
2014/05/06 职场文书
自我管理的活动方案
2014/08/25 职场文书
2014年保洁工作总结
2014/11/24 职场文书
初婚初育证明范本
2015/06/18 职场文书