在TensorFlow中实现矩阵维度扩展
一般TensorFlow中扩展维度可以使用tf.expand_dims()。近来发现另一种可以直接运用取数据操作符[]就能扩展维度的方法。
用法很简单，在要扩展的维度上加上tf.newaxis就行了。
foo = tf.constant([[1,2,3], [4,5,6], [7,8,9]]) print(foo[tf.newaxis, :, :].eval()) # => [[[1,2,3], [4,5,6], [7,8,9]]] print(foo[:, tf.newaxis, :].eval()) # => [[[1,2,3]], [[4,5,6]], [[7,8,9]]] print(foo[:, :, tf.newaxis].eval()) # => [[[1],[2],[3]], [[4],[5],[6]],[[7],[8],[9]]]
参考：
https://tensorflow.google.cn/api_docs/python/tf/Tensor?hl=en#__getitem__
补充知识：tensorflow 利用expand_dims和squeeze扩展和压缩tensor维度
在利用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中实现矩阵维度扩展就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
使用Python的Tornado框架实现一个Web端图书展示页面
Tornado是Python的一款高人气Web开发框架,这里我们来展示使用Python的Tornado框架实现一个Web端图书展示页面的实例,通过该实例可以清楚地学习到Tornado的模板使用及整个Web程序的执行流程.2016-07-07
本篇文章主要介绍了python 实现A*算法的示例代码，A*作为最常用的路径搜索算法，值得我们去深刻的研究，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-08-08
今天小编就为大家分享一篇Python开启线程,在函数中开线程的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-02-02
如果能够将我们的无序数据快速组织成更易读的格式,对于数据分析非常有帮助,下面这篇文章主要介绍了关于如何使用Python代码创建表格的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下2023-06-06
本文主要介绍了Python10行代码实现模拟百度搜索的示例，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-07-07
这篇文章主要介绍了基于Django signals 信号作用及用法详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-03-03
这篇文章主要是带着大家通过Python Tkinter制作一个属于自己的GUI图形化界面，可以用于设计签名的哦，感兴趣的小伙伴快跟随小编一起学习一下吧2022-04-04
Rich库的功能就像它的名字一样,使Python编程更加丰富（rich）,用来帮助开发者在控制台（命令行）输出中创建丰富、多彩和具有格式化的文本,下面我们就来了解下它的具体使用吧2024-02-02
这篇文章主要为大家详细介绍了Ubuntu18.04下python版本完美切换的解决方法，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-06-06
Python捕获全局的KeyboardInterrupt异常的方法实现
KeyboardInterrupt异常是Python中的一个标准异常,它通常发生在用户通过键盘中断了一个正在运行的程序,本文主要介绍了Python捕获全局的KeyboardInterrupt异常的方法实现,感兴趣的可以了解一下2024-08-08
最新评论