[Python知识库]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维度，对应下面图片中的红框部分