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

作者:胡大炮的妖孽人生 时间:2023-08-25 21:55:40 

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:增加或者减小矩阵维度的实现

来源:https://blog.csdn.net/huplion/article/details/80093292

标签:Python3,Tensorlfow,矩阵,维度
0
投稿

猜你喜欢

  • SQLServer 2008 新增T-SQL 简写语法

    2024-01-28 23:49:25
  • 无法远程登入MySQL数据库的几种解决办法

    2012-04-13 11:57:41
  • Python图片视频超分模型RealBasicVSR的使用教程

    2021-10-11 13:40:04
  • WAP2.0网页设计中的交互细节

    2010-08-12 20:50:00
  • 使用typescript快速开发一个cli的实现示例

    2023-08-30 07:25:25
  • js 创建一个浮动div的代码

    2024-04-10 14:02:55
  • 对Python函数设计规范详解

    2023-08-02 15:59:17
  • Python requests HTTP验证登录实现流程

    2021-02-27 16:43:04
  • python实现12306抢票及自动邮件发送提醒付款功能

    2021-03-12 19:01:46
  • flask-socketio实现前后端实时通信的功能的示例

    2021-02-18 22:18:35
  • golang开发go包依赖管理godep使用教程

    2023-10-16 04:09:31
  • MySQL使用ReplicationConnection导致连接失效解决

    2024-01-19 11:04:26
  • Python设计模式之备忘录模式原理与用法详解

    2022-12-27 11:12:01
  • oracle数据库创建备份与恢复脚本整理

    2023-07-13 00:57:20
  • 对抗MySQL数据库解密高手

    2008-12-25 13:14:00
  • JavaScript数字和字符串转换示例

    2024-05-02 16:27:07
  • node实现mock-plugin中间件的方法

    2024-05-13 10:05:59
  • 原生javascript AJAX 三级联动的实现代码

    2024-04-18 10:00:46
  • DreamWeaver制作会移动的广告条

    2008-02-03 11:34:00
  • 利用css的clear属性搞定广告文字环绕效果

    2008-05-24 13:48:00
  • asp之家 网络编程 m.aspxhome.com