tensorflow 利用expand_dims和squeeze扩展和压缩tensor维度方式


Posted in Python onFebruary 07, 2020

在利用tensorflow进行文本挖掘工作的时候,经常涉及到维度扩展和压缩工作。比如对文本进行embedding操作完成之后,若要进行卷积操作,就需要对embedded的向量扩展维度,将[batch_size, embedding_dims]扩展成为[batch_size, embedding_dims, 1],利用tf.expand_dims(input, -1)就可实现,反过来用squeeze(input, -1)或者tf.squeeze(input)也可以把最第三维去掉。

tf.expand_dims()

tf.squeeze()

tf.expand_dims()

tf.expand_dims(input, axis=None, name=None, dim=None)

在第axis位置增加一个维度.

给定张量输入,此操作在输入形状的维度索引轴处插入1的尺寸。 尺寸索引轴从零开始; 如果您指定轴的负数,则从最后向后计数。

如果要将批量维度添加到单个元素,则此操作非常有用。 例如,如果您有一个单一的形状[height,width,channels],您可以使用expand_dims(image,0)使其成为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]

tf.squeeze()

tf.squeeze(input, axis=None, name=None, squeeze_dims=None)

直接上例子

# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
 shape(squeeze(t)) ==> [2, 3]
# 't' is a tensor of shape [1, 2, 1, 3, 1, 1]
 shape(squeeze(t, [2, 4])) ==> [1, 2, 3, 1]

以上这篇tensorflow 利用expand_dims和squeeze扩展和压缩tensor维度方式就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python实现文件名批量替换和内容替换
Mar 20 Python
Python contextlib模块使用示例
Feb 18 Python
简单的连接MySQL与Python的Bottle框架的方法
Apr 30 Python
Python使用bs4获取58同城城市分类的方法
Jul 08 Python
Python Socket编程详细介绍
Mar 23 Python
Python根据欧拉角求旋转矩阵的实例
Jan 28 Python
Django框架实现的普通登录案例【使用POST方法】
May 15 Python
Python使用字典实现的简单记事本功能示例
Aug 15 Python
Python使用Beautiful Soup爬取豆瓣音乐排行榜过程解析
Aug 15 Python
带你彻底搞懂python操作mysql数据库(cursor游标讲解)
Jan 06 Python
Tensorflow实现在训练好的模型上进行测试
Jan 20 Python
Python虚拟环境virtualenv创建及使用过程图解
Dec 08 Python
Tensorflow进行多维矩阵的拆分与拼接实例
Feb 07 #Python
Tensorflow训练模型越来越慢的2种解决方案
Feb 07 #Python
详解python itertools功能
Feb 07 #Python
Python中itertools的用法详解
Feb 07 #Python
Python转换itertools.chain对象为数组的方法
Feb 07 #Python
已安装tensorflow-gpu,但keras无法使用GPU加速的解决
Feb 07 #Python
python十进制转二进制的详解
Feb 07 #Python
You might like
浅析PHP的静态成员函数效率更高的原因
2014/06/13 PHP
php中unserialize返回false的解决方法
2014/09/22 PHP
使用PHP进行微信公众平台开发的示例
2015/08/21 PHP
分享3个php获取日历的函数
2015/09/25 PHP
PHP文件下载实例代码浅析
2016/08/17 PHP
js实时获取并显示当前时间的方法
2015/07/31 Javascript
BootStrap学习系列之Bootstrap Typeahead 组件实现百度下拉效果(续)
2016/07/07 Javascript
node.js中express中间件body-parser的介绍与用法详解
2017/05/23 Javascript
详解在vue-cli项目中安装node-sass
2017/06/21 Javascript
基于JS实现html中placeholder属性提示文字效果示例
2018/04/19 Javascript
layer弹出的iframe层在执行完毕后关闭当前弹出层的方法
2018/08/17 Javascript
jQuery实现的模仿雨滴下落动画效果
2018/12/11 jQuery
vue 表单验证按钮事件交由父组件触发的方法
2018/12/17 Javascript
基于form-data请求格式详解
2019/10/29 Javascript
vue 实现v-for循环回来的数据动态绑定id
2019/11/07 Javascript
微信小程序纯文本实现@功能
2020/04/08 Javascript
node使用async_hooks模块进行请求追踪
2021/01/28 Javascript
[58:54]EG vs RNG 2019国际邀请赛小组赛 BO2 第一场 8.16
2019/08/18 DOTA
Python中的Django基本命令实例详解
2018/07/15 Python
Python使用reportlab模块生成PDF格式的文档
2019/03/11 Python
python 在某.py文件中调用其他.py内的函数的方法
2019/06/25 Python
python实现简单银行管理系统
2019/10/25 Python
在Tensorflow中实现leakyRelu操作详解(高效)
2020/06/30 Python
django form和field具体方法和属性说明
2020/07/09 Python
python中如何使用虚拟环境
2020/10/14 Python
CSS3 Media Queries(响应式布局可以让你定制不同的分辨率和设备)
2013/06/06 HTML / CSS
CSS3关于z-index不生效问题的解决
2020/02/19 HTML / CSS
售后主管岗位职责
2013/12/08 职场文书
财务会计人员求职的自我评价
2014/01/13 职场文书
医生进修自我鉴定
2014/01/19 职场文书
特教教师先进事迹
2014/05/21 职场文书
2015年春训学习心得体会范文
2015/03/09 职场文书
趣味运动会简讯
2015/07/20 职场文书
2016高一新生军训心得体会
2016/01/11 职场文书
「我的青春恋爱物语果然有问题。-妄言录-」第20卷封面公开
2022/03/21 日漫
Android移动应用开发指南之六种布局详解
2022/09/23 Java/Android