pytorch中squeeze()、unsqueeze(),以及一些高维数组操作

妖狐艹你老母 2024-04-18 18:19 60阅读 0赞

博主最近阅读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  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl8zODYzMjI0Ng_size_16_color_FFFFFF_t_70]

### 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)

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl8zODYzMjI0Ng_size_16_color_FFFFFF_t_70 1]

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

注意：b的维度比a少了一维。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl8zODYzMjI0Ng_size_16_color_FFFFFF_t_70 2]继续上一步，这里使用unsqueeze函数将b的维度扩展一维\[2,2,3\]------>\[2,2,3,1\] 这时b的最后一维只有一个元素， .expand\_as将最后最后的元素扩展到a最后一维的个数

a\[c\] 表示取出为True的所有行在a中的元素  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl8zODYzMjI0Ng_size_16_color_FFFFFF_t_70 3]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl8zODYzMjI0Ng_size_16_color_FFFFFF_t_70]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/18/21a719b81ebf4c5f864fe1f6674e4282.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl8zODYzMjI0Ng_size_16_color_FFFFFF_t_70 1]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/18/c08b7d736b0843aaaceea21d8717e72e.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl8zODYzMjI0Ng_size_16_color_FFFFFF_t_70 2]: https://img-blog.csdnimg.cn/20190910151139907.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl8zODYzMjI0Ng==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl8zODYzMjI0Ng_size_16_color_FFFFFF_t_70 3]: https://image.dandelioncloud.cn/pgy_files/images/2024/04/18/8a2ae65673ff4347a3d13d0d0a11cfcd.png