Python3 Tensorlfow:增加或者减小矩阵维度的实现


Posted in Python onMay 22, 2020

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) 会改变

Python3 Tensorlfow:增加或者减小矩阵维度的实现

这里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)

Python3 Tensorlfow:增加或者减小矩阵维度的实现

看一个简单用例,size表示维度大小,10是取值范围,a=[:,:,:,4]表示取a最后一维的第四个元素(从0开始第四个),即取[0,0,3],[5,6,1],[0,6,8],[…], 判断大于5为true,否则为false。

注意:b的维度比a少了一维。

Python3 Tensorlfow:增加或者减小矩阵维度的实现

继续上一步,这里使用unsqueeze函数将b的维度扩展一维[2,2,3]------>[2,2,3,1] 这时b的最后一维只有一个元素, .expand_as将最后最后的元素扩展到a最后一维的个数

a[c] 表示取出为True的所有行在a中的元素

Python3 Tensorlfow:增加或者减小矩阵维度的实现

以上这篇Python3 Tensorlfow:增加或者减小矩阵维度的实现就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python实现爬虫下载漫画示例
Feb 16 Python
浅析python中SQLAlchemy排序的一个坑
Feb 24 Python
Python双精度浮点数运算并分行显示操作示例
Jul 21 Python
tensorflow 用矩阵运算替换for循环 用tf.tile而不写for的方法
Jul 27 Python
【python】matplotlib动态显示详解
Apr 11 Python
Python threading的使用方法解析
Aug 28 Python
django框架F&Q 聚合与分组操作示例
Dec 12 Python
Python 读取有公式cell的结果内容实例方法
Feb 17 Python
Pytest参数化parametrize使用代码实例
Feb 22 Python
Keras - GPU ID 和显存占用设定步骤
Jun 22 Python
python 使用xlsxwriter循环向excel中插入数据和图片的操作
Jan 01 Python
Spy++的使用方法及下载教程
Jan 29 Python
Python pandas如何向excel添加数据
May 22 #Python
python3.8.1+selenium实现登录滑块验证功能
May 22 #Python
在TensorFlow中实现矩阵维度扩展
May 22 #Python
Python多个装饰器的调用顺序实例解析
May 22 #Python
基于Python第三方插件实现西游记章节标注汉语拼音的方法
May 22 #Python
Tensorflow tf.tile()的用法实例分析
May 22 #Python
python代码实现将列表中重复元素之间的内容全部滤除
May 22 #Python
You might like
全国FM电台频率大全 - 7 吉林省
2020/03/11 无线电
树型结构列出指定目录里所有文件的PHP类
2006/10/09 PHP
php中定时计划任务的实现原理
2013/01/08 PHP
php实现查看邮件是否已被阅读的方法
2013/12/03 PHP
总结PHP删除字符串最后一个字符的三种方法
2016/08/30 PHP
浅谈PHP发送HTTP请求的几种方式
2017/07/25 PHP
使用JavaScript检测Firefox浏览器是否启用了Firebug的代码
2010/12/28 Javascript
写自已的js类库需要的核心代码
2012/07/16 Javascript
ajax页面无刷新 IE下遭遇Ajax缓存导致数据不更新的问题
2012/12/11 Javascript
js获取url参数值的两种方式
2013/09/10 Javascript
基于jQuery实现的幻灯图片切换
2016/12/02 Javascript
nodejs中art-template模板语法的引入及冲突解决方案
2017/11/07 NodeJs
基于layui table返回的值的多级嵌套的解决方法
2019/09/19 Javascript
通过实例了解JS 连续赋值
2019/09/24 Javascript
js实现左右轮播图
2020/01/09 Javascript
[01:48]DOTA2 2015国际邀请赛中国区预选赛第二日战报
2015/05/27 DOTA
[40:55]Liquid vs LGD 2018国际邀请赛小组赛BO2 第二场 8.16
2018/08/17 DOTA
easy_install python包安装管理工具介绍
2013/02/10 Python
python赋值操作方法分享
2013/03/23 Python
c++生成dll使用python调用dll的方法
2014/01/20 Python
python里大整数相乘相关技巧指南
2014/09/12 Python
八大排序算法的Python实现
2021/01/28 Python
python奇偶行分开存储实现代码
2018/03/19 Python
详解将Django部署到Centos7全攻略
2018/09/26 Python
使用Python和Prometheus跟踪天气的使用方法
2019/05/06 Python
python实现简单日志记录库glog的使用
2019/12/13 Python
python颜色随机生成器的实例代码
2020/01/10 Python
Python爬虫代理池搭建的方法步骤
2020/09/28 Python
中国包裹转运寄送国际服务:Famiboat
2019/07/24 全球购物
Stokke美国官方网店:高级儿童家具、推车、汽车座椅和配件
2020/06/06 全球购物
美国家居装饰购物网站:Amanda Lindroth
2020/03/25 全球购物
《维生素c的故事》教学反思
2014/02/18 职场文书
大学生档案自我鉴定(2篇)
2014/10/14 职场文书
幼儿园老师新年寄语2015
2014/12/08 职场文书
兴趣班停课通知
2015/04/24 职场文书
太空授课观后感
2015/06/17 职场文书