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中list常用操作实例详解
Jun 03 Python
用python写一个windows下的定时关机脚本(推荐)
Mar 21 Python
Python3 模块、包调用&路径详解
Oct 25 Python
Python学习小技巧总结
Jun 10 Python
Python如何爬取实时变化的WebSocket数据的方法
Mar 09 Python
浅析Python3中的对象垃圾收集机制
Jun 06 Python
python django 原生sql 获取数据的例子
Aug 14 Python
Python Web框架之Django框架文件上传功能详解
Aug 16 Python
如何通过python实现人脸识别验证
Jan 17 Python
基于Python实现视频的人脸融合功能
Jun 12 Python
python缺失值的解决方法总结
Jun 09 Python
python数字图像处理实现图像的形变与缩放
Jun 28 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
example2.php
2006/10/09 PHP
PHP基本语法总结
2014/09/06 PHP
PHP中auto_prepend_file与auto_append_file用法实例分析
2014/09/22 PHP
php libevent 功能与使用方法详解
2020/03/04 PHP
jQuery toggle()设置CSS样式
2009/11/05 Javascript
JavaScript自定义事件介绍
2013/08/29 Javascript
JS与C#编码解码
2013/12/03 Javascript
在Firefox下js select标签点击无法弹出
2014/03/06 Javascript
jQuery中:lt选择器用法实例
2014/12/29 Javascript
使用ngView配合AngularJS应用实现动画效果的方法
2015/06/19 Javascript
javascript中checkbox使用方法实例演示
2015/11/19 Javascript
ES6中箭头函数的定义与调用方式详解
2017/06/02 Javascript
js获取html页面代码中图片地址的实现代码
2018/03/05 Javascript
Vue.js 实现微信公众号菜单编辑器功能(一)
2018/05/08 Javascript
如何用JS模拟实现数组的map方法
2020/07/30 Javascript
[03:24]DOTA2超级联赛专访hao 大翻盘就是逆袭
2013/05/24 DOTA
[02:43]2014DOTA2国际邀请赛 官方Alliance战队纪录片
2014/07/14 DOTA
在Python的web框架中中编写日志列表的教程
2015/04/30 Python
在Python中用keys()方法返回字典键的教程
2015/05/21 Python
Python实现求笛卡尔乘积的方法
2017/09/16 Python
Python3使用PyQt5制作简单的画板/手写板实例
2017/10/19 Python
Python set常用操作函数集锦
2017/11/15 Python
python自定义线程池控制线程数量的示例
2019/02/22 Python
Python TestCase中的断言方法介绍
2019/05/02 Python
python 已知一个字符,在一个list中找出近似值或相似值实现模糊匹配
2020/02/29 Python
python 实现仿微信聊天时间格式化显示的代码
2020/04/17 Python
英国领先的男装设计师服装购物网站:Mainline Menswear
2018/02/04 全球购物
微软美国官方网站:Microsoft美国
2018/05/10 全球购物
护理专科毕业生自荐书范文
2014/02/19 职场文书
我有一个梦想演讲稿
2014/05/05 职场文书
秋季运动会加油词
2015/07/18 职场文书
寒假致家长的一封信
2015/10/10 职场文书
小学四年级作文之写景
2019/08/23 职场文书
python 统计代码耗时的几种方法分享
2021/04/02 Python
Python如何让字典保持有序排列
2022/04/29 Python
mysql5.5中文乱码问题解决的有用方法
2022/05/30 MySQL