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)
还没有任何评论哟~
