tensorflow 利用expand_dims和squeeze扩展和压缩tensor维度方式
作者:LauJames 时间:2023-04-25 04:15:56
在利用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]
来源:https://blog.csdn.net/UESTC_V/article/details/80310487
标签:tensorflow,expand,dims,squeeze,tensor
0
投稿
猜你喜欢
SQLServer2005触发器提示其他会话正在使用事务的上下文的解决方法
2024-01-13 05:42:28
关于vue父组件调用子组件的方法
2024-04-09 10:47:43
vue element-ul实现展开和收起功能的实例代码
2024-05-29 22:48:16
mysql常用备份命令和shell备份脚本分享
2024-01-13 14:37:35
Python操作SQLite数据库的方法详解【导入,创建,游标,增删改查等】
2024-01-19 09:29:00
python判断字符串是否纯数字的方法
2022-09-09 03:25:38
go语言csrf库使用实现原理示例解析
2023-08-07 03:34:38
对numpy Array [: ,] 的取值方法详解
2023-02-05 17:33:23
胜过语言的图形符号
2009-05-06 12:43:00
python指定路径斜杠与反斜杠遇到的问题
2023-03-18 20:47:27
Springboot如何同时装配两个相同类型数据库
2024-01-25 08:51:33
PHP7正式版测试,性能惊艳!
2023-09-12 07:41:43
Python 多线程处理任务实例
2021-06-25 04:50:05
在django admin中配置搜索域是一个外键时的处理方法
2021-02-20 18:48:26
Python变量格式化输出实现原理解析
2023-02-12 23:13:51
pytorch_pretrained_bert如何将tensorflow模型转化为pytorch模型
2022-04-18 18:07:58
详解python tcp编程
2021-04-30 00:12:46
yolov5训练时参数workers与batch-size的深入理解
2021-08-01 04:50:20
计算机程序设计并行计算概念及定义全面详解
2023-02-20 22:35:59
node+axios实现下载外网文件到本地
2024-05-05 09:20:48