Python3 Tensorlfow:增加或者减小矩阵维度的实现
1.增加维度
下面给出两个样例
样例1：
[1, 2, 3] ==> [[1],[2],[3]]
import tensorflow as tf
a = tf.constant([1, 2, 3])
b = tf.expand_dims(a,1)
with tf.Session() as sess:
a_, b_ = sess.run([a, b])
print('a:')
print(a_)
print('b:')
print(b_)
输出结果
a: [1 2 3] b: [[1] [2] [3]]
样例2:
[1, 2, 3] ==> [[1,2,3]]
import tensorflow as tf
a = tf.constant([1, 2, 3])
b = tf.expand_dims(a, 0)
with tf.Session() as sess:
a_, b_ = sess.run([a, b])
print('a:')
print(a_)
print('b:')
print(b_)
输出结果：
a: [1 2 3] b: [[1 2 3]]
2.降低维度
样例1：
[[1, 2, 3]] ==> [1, 2, 3]
import tensorflow as tf
a = tf.constant([[1, 2, 3]])
b = tf.squeeze(a)
with tf.Session() as sess:
a_, b_ = sess.run([a, b])
print('a:')
print(a_)
print('b:')
print(b_)
输出结果
a: [[1 2 3]] b: [1 2 3]
样例2：
[[1], [2], [3]] ==> [[1, 2, 3]
import tensorflow as tf
a = tf.constant([[1], [2], [3]])
b = tf.squeeze(a, 1)
with tf.Session() as sess:
a_, b_ = sess.run([a, b])
print('a:')
print(a_)
print('b:')
print(b_)
补充知识：pytorch中squeeze()、unsqueeze(),以及一些高维数组操作
博主最近阅读YOLO底层代码，Torch中对多数组矩阵有很多高维操作，看过一边之后，记录一下，以防忘记。
torch.squeeze()
功能：取消为1的维度
squeeze(input, dim=None, out=None) -> Tensor
这里一般分不清dim的意思
举个例子：
input=(A ， 1 ， B ， C ，1 ， D) squeeze(input)=(A，B，C，D) input= （A, 1, B）
squeeze(input, 0)=(A, 1, B) 不会改变 squeeze(input, 1)=(A, B) 会改变
这里0, 1 ，2就分别代表 A， 1, B
torch.unsqueeze()
unsqueeze(input, dim, out=None) -> Tensor
功能： 在某一位置插入一维
这里面也是dim这个参数比较难理解
dim的取值是[- input.dim()-1, imput.dim()]
给定一个dim
input=(A ， B ， C ， D)
input的维度input_dim为4, dim的取值[-5, 4]
unsqueeze(input, 0)=(1, A ， B ， C ， D) unsqueeze(input, 1)=(A ， 1, B ， C ， D) unsqueeze(input, -5)=(1, A ， B ， C ， D)
看一个简单用例，size表示维度大小，10是取值范围，a=[：，：，：，4]表示取a最后一维的第四个元素（从0开始第四个），即取[0,0,3],[5,6,1],[0,6,8],[…], 判断大于5为true，否则为false。
注意：b的维度比a少了一维。
继续上一步，这里使用unsqueeze函数将b的维度扩展一维[2,2,3]------>[2,2,3,1] 这时b的最后一维只有一个元素， .expand_as将最后最后的元素扩展到a最后一维的个数
a[c] 表示取出为True的所有行在a中的元素
以上这篇Python3 Tensorlfow:增加或者减小矩阵维度的实现就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
这篇文章主要介绍了详解Python 4.0 预计推出的新功能，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2019-07-07
Selenium是一个用于Web应用程序测试的工具。Selenium测试直接运行在浏览器中，就像真正的用户在操作一样。本文主要介绍了利用Selenium实现自动观看学习通视频，需要的同学可以参考一下2021-12-12
这篇文章主要介绍了Python扫描IP段查看指定端口是否开放的方法,涉及Python使用socket模块实现端口扫描功能的相关技巧,需要的朋友可以参考下2015-06-06
django框架model orM使用字典作为参数，保存数据的方法分析
这篇文章主要介绍了django框架model orM使用字典作为参数，保存数据的方法,结合实例形式分析了字典参数结合django model实现保存数据相关操作技巧,需要的朋友可以参考下2019-06-06
这篇文章主要介绍了Python 中导入csv数据的三种方法,内容比较简单，非常不错，具有一定的参考借鉴价值，需要的朋友可以参考下2018-11-11
这篇文章主要介绍了Python网络通讯之TCP协议实现服务器和客户端实例, socket编程一种独立于协议的网络编程接口，应用程序可以通过它发送或接收数据，可对其进行像对文件一样的打开、读写和关闭等操作,需要的朋友可以参考下2023-08-08
本文主要介绍了浅谈Django Admin的初步使用 ，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2021-12-12
这篇文章主要介绍了python cv2.waitKey()函数的使用方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2024-01-01
今天小编就为大家分享一篇python指定写入文件时的编码格式方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-06-06
这篇文章主要介绍了python爬虫之你好,李焕英电影票房数据分析,文中有非常详细的代码示例,对正在学习python爬虫的小伙伴们有一定的帮助,需要的朋友可以参考下2021-04-04
最新评论