Advertisement

torch.unsqueeze()函数的解析内容来自《PyTorch学习》

阅读量:

引言:该函数在进行维度计算时被广泛使用,特别是在网络构建过程中。笔者在此仅作初步理解,将其视为一种能够增加维度的函数,即在原有维度基础上额外扩展出一个一维空间。

若对引言部分尚存困惑,接下来将通过一些简单明了的例子进行说明。(通常情况下,图像的表示方式为(CWH),其中C表示通道数量,即常见的三原色RGB通道,W代表图像的宽度,H代表图像的高度,以下将以这种方式进行讲解)

数据库导入:

复制代码
>       1. import torch

>  
>       2. from PIL import Image
>  
>       3. import matplotlib.pyplot as plt
>  
>  
>  
>  
> ```

>
>
>
> 1.首先在一维空间的张量结构中,于起始位置新增/扩展一个维度。(Y,X)
>
>
>
> 1》沿Y轴方向进行维度扩展
>
>
>
> 代码:
>
>
复制代码
  1. data1=torch.tensor([1,2,5,9])
复制代码
  2. print(data1,data1.dim(),data1.size())

  3. data2=torch.unsqueeze(data1,0)

全部评论 (0)

还没有任何评论哟~