本文介绍了在PyTorch中如何使用squeeze和unsqueeze函数来处理数组的维度。在numpy中，有时会遇到秩为1的一维数组，它们既非行向量也非列向量。squeeze函数用于删除尺寸为1的轴，而unsqueeze则在指定位置插入尺寸为1的轴。通过这两个函数，可以灵活地调整数组的形状，以适应不同的运算需求。
在numpy库中，经常会出现“秩为1的一维数组”（come from 吴恩达的深度学习，目前还没有搞清楚numpy中如此设计的意图）。比如：
>>> a = torch.rand(3)
>>> a.shape
torch.Size([3])
注意这里的a的shape是[3] ，既不是 [1,3] 也不是 [3,1]。这就说明它既不是行向量也不是列向量，只是一个数组。
但是我们可以用squeeze（）和unsqueeze（）对其进行操作，比如：
>>> a = torch.rand(3)
>>> a.shape
torch.Size([3])
>>> b = a.unsqueeze(1)
>>> b.shape
torch.Size([3
最低0.47元/天 解锁文章
5369
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
