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 相关文章推荐
Python列表和元组的定义与使用操作示例
Jul 26 Python
Python基于pygame模块播放MP3的方法示例
Sep 30 Python
python2.7 json 转换日期的处理的示例
Mar 07 Python
分享一下Python数据分析常用的8款工具
Apr 29 Python
python看某个模块的版本方法
Oct 16 Python
python中yield的用法详解——最简单,最清晰的解释
Apr 04 Python
python基于paramiko将文件上传到服务器代码实现
Jul 08 Python
python实现简单成绩录入系统
Sep 19 Python
Python图像处理库PIL的ImageDraw模块介绍详解
Feb 26 Python
Python sorted排序方法如何实现
Mar 31 Python
Python 合并拼接字符串的方法
Jul 28 Python
Python制作春联的示例代码
Jan 22 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
全国FM电台频率大全 - 6 辽宁省
2020/03/11 无线电
Laravel统计一段时间间隔的数据方法
2019/10/09 PHP
有关PHP 中 config.m4 的探索
2020/08/26 PHP
jquery 模拟类搜索框自动完成搜索提示功能(改进)
2010/05/24 Javascript
JQuery Ajax通过Handler访问外部XML数据的代码
2010/06/01 Javascript
IE6、IE7中setAttribute不支持class/for/rowspan/colspan等属性
2011/08/28 Javascript
使用Javascript接收get传递的值的代码
2011/11/30 Javascript
转换字符串为json对象的方法详解
2013/11/29 Javascript
js showModalDialog参数的使用详解
2014/01/07 Javascript
select多选 multiple的使用示例
2014/06/16 Javascript
JavaScript中原型和原型链详解
2015/02/11 Javascript
黑帽seo劫持程序,js劫持搜索引擎代码
2015/09/15 Javascript
易操作的jQuery表单提示插件
2015/12/01 Javascript
原生js实现无限循环轮播图效果
2017/01/20 Javascript
基于vue实现多引擎搜索及关键字提示
2017/03/16 Javascript
详解vue.js全局组件和局部组件
2017/04/10 Javascript
Vue中计算属性computed的示例解读
2017/07/26 Javascript
js图片上传的封装代码
2017/08/01 Javascript
vue实现微信获取用户信息的方法
2019/03/21 Javascript
NodeJs操作MongoDB教程之分页功能以及常见问题
2019/04/09 NodeJs
vuex actions异步修改状态的实例详解
2019/11/06 Javascript
JavaScript冒泡算法原理与实现方法深入理解
2020/06/04 Javascript
js 将多个对象合并成一个对象 assign方法的实现
2020/09/24 Javascript
Python中map,reduce,filter和sorted函数的使用方法
2015/08/17 Python
浅谈python中的面向对象和类的基本语法
2016/06/13 Python
python实现定时压缩指定文件夹发送邮件
2020/12/22 Python
Python使用enumerate获取迭代元素下标
2020/02/03 Python
手把手教你将Flask应用封装成Docker服务的实现
2020/08/19 Python
python连接mysql数据库并读取数据的实现
2020/09/25 Python
乡镇庆八一活动方案
2014/02/02 职场文书
企业党员公开承诺书
2014/03/26 职场文书
感恩寄语大全
2014/04/11 职场文书
幼儿园评语大全
2014/04/17 职场文书
中学生运动会口号
2014/06/07 职场文书
师范生教育见习总结
2015/06/23 职场文书
介绍信应该怎么开?
2019/04/03 职场文书