python以三维tensor为例详细理解unsqueeze和squeeze函数

简介: python以三维tensor为例详细理解unsqueeze和squeeze函数

一、建立一个二维tensor

import torch as t
a = t.arange(0,6)
b = a.view(2,3)
print(b)

二、对上步建立的二维tensor使用unsqueeze(x)函数操作(x若是正数表示在第x维上增加一维,若是负数表示在倒数第x维上增加一维,注意tensor维度的下标从0开始,倒数下标从-1开始,并且此增加维的维度为1)

2.1 x为正数

import torch as t
a = t.arange(0,6)
b = a.view(2,3)
print(b.size())
b = t.unsqueeze(b,1) # unsqueeze函数操作,在第一维上增加“1”
print(b.size())

2.2 x为负数

import torch as t
a = t.arange(0,6)
b = a.view(2,3)
print(b.size())
b = t.unsqueeze(b,-1) # unsqueeze函数操作,在倒数第一维上增加“1”
print(b.size())

三、对第一步建立的二维tensor使用squeeze(x)函数操作(x若为正数表示压缩第x维的“1”,若是负数表示压缩倒数第x维的“1”,若为null则压缩所有维的“1”,若第x维的维度不为“1”就无变化不能压缩,注意tensor维度的下标从0开始,倒数下标从-1开始)

3.1 x为正数

import torch as t
a = t.arange(0,6)
b = a.view(1,1,2,3,1)
print(b.size())
b = t.squeeze(b,1) # squeeze函数操作,在压缩第一维的“1”
print(b.size())

3.2 x为负数

import torch as t
a = t.arange(0,6)
b = a.view(1,1,2,3,1)
print(b.size())
b = t.squeeze(b,-1) # squeeze函数操作,压缩倒数第一维的“1”
print(b.size())

3.3 x为null(空)

import torch as t
a = t.arange(0,6)
b = a.view(1,1,2,3,1)
print(b.size())
b = t.squeeze(b) # squeeze函数操作,压缩所有维度为“1”的维
print(b.size())

3.4 x维的维度不等于“1”

import torch as t
a = t.arange(0,6)
b = a.view(1,1,2,3,1)
print(b.size())
b = t.squeeze(b,3) # squeeze函数操作,第3维的维度是“3”,不是“1”,所以不能压缩无变化
print(b.size())

四、多维具体理解分析(以三维为例)

4.1 二维变成三维的代码

import torch as t
a = t.arange(0,6)
b = a.view(3,2)
print(b)
b = t.unsqueeze(b,1)
print(b)

4.2 代码运行输出结果

4.3 结果理解分析

tensor([[0, 1],
        [2, 3],
        [4, 5]])
tensor([[[0, 1]],
        [[2, 3]],
        [[4, 5]]])

原本tensor是[3,2]也就是三行两列,在使用unsqueeze函数后增加第一维的维度为“1”,也就变成了[3,1,2]也就是312变成三维tensor,上面的结果对应312的理解是tensor()里面第一层[ ]中有三行,并且用逗号隔开对应“3”,第一层[ [ ] , [ ] , [ ] ]里面的每个分开的[ ]也就是第二层[ ]里面是只有一行,因为只有一行就没有用逗号隔开对应“1”,第二层[ ]里面的[ ]也就是第三层[ ]里面有两列也就是两个元素对应“2”。

4.4 tensor维度的直接显示

import torch as t
a = t.arange(0,6)
b = a.view(3,2)
print(b)
b = t.unsqueeze(b,1)
print(b)
print(b.size()) # 直接输出显示tensor维度,对应下面图片中的红框部分


相关文章
|
1月前
|
Python
【python从入门到精通】-- 第五战:函数大总结
【python从入门到精通】-- 第五战:函数大总结
63 0
|
28天前
|
Python
Python之函数详解
【10月更文挑战第12天】
Python之函数详解
|
29天前
|
存储 数据安全/隐私保护 索引
|
18天前
|
测试技术 数据安全/隐私保护 Python
探索Python中的装饰器:简化和增强你的函数
【10月更文挑战第24天】在Python编程的海洋中,装饰器是那把可以令你的代码更简洁、更强大的魔法棒。它们不仅能够扩展函数的功能,还能保持代码的整洁性。本文将带你深入了解装饰器的概念、实现方式以及如何通过它们来提升你的代码质量。让我们一起揭开装饰器的神秘面纱,学习如何用它们来打造更加优雅和高效的代码。
|
20天前
|
弹性计算 安全 数据处理
Python高手秘籍:列表推导式与Lambda函数的高效应用
列表推导式和Lambda函数是Python中强大的工具。列表推导式允许在一行代码中生成新列表,而Lambda函数则是用于简单操作的匿名函数。通过示例展示了如何使用这些工具进行数据处理和功能实现,包括生成偶数平方、展平二维列表、按长度排序单词等。这些工具在Python编程中具有高度的灵活性和实用性。
|
23天前
|
Python
python的时间操作time-函数介绍
【10月更文挑战第19天】 python模块time的函数使用介绍和使用。
27 4
|
24天前
|
存储 Python
[oeasy]python038_ range函数_大小写字母的起止范围_start_stop
本文介绍了Python中`range`函数的使用方法及其在生成大小写字母序号范围时的应用。通过示例展示了如何利用`range`和`for`循环输出指定范围内的数字,重点讲解了小写和大写字母对应的ASCII码值范围,并解释了`range`函数的参数(start, stop)以及为何不包括stop值的原因。最后,文章留下了关于为何`range`不包含stop值的问题,留待下一次讨论。
18 1
|
30天前
|
索引 Python
Python中的其他内置函数有哪些
【10月更文挑战第12天】Python中的其他内置函数有哪些
15 1
|
24天前
|
安全 数据处理 数据安全/隐私保护
python中mod函数怎么用
通过这些实例,我们不仅掌握了Python中 `%`运算符的基础用法,还领略了它在解决实际问题中的灵活性和实用性。在诸如云计算服务提供商的技术栈中,类似的数学运算逻辑常被应用于数据处理、安全加密等关键领域,凸显了基础运算符在复杂系统中的不可或缺性。
18 0
|
30天前
|
开发者 索引 Python
Python中有哪些内置函数
【10月更文挑战第12天】Python中有哪些内置函数
18 0