pytorch每日一学47(torch.unsqueeze())在指定维度上增加一个大小为1的维度

第47个方法

torch.unsqueeze(input, dim) → Tensor

很明显这个方法是与
torch.squeeze()进行相反的操作，torch.squeeze()是将移出tensor中维度为1的维度，而本次的这个方法就是增加一个维度，大小为1。
如下所示：

指定了给定维度以后，就会在指定维度增加一个维度，大小为1。其实也就是将那个维度上的直接拆开。如果为1的话：

如果为0的话则和之前一样，因为0维度本来就只有1个。而本例中如果超过2的话会报错：

所以指定维度不能超过给定tensor的维度。

当然可以取负数，其实负数就是从最后一个开始，例如-1等于2，效果一样，如下所示：

其它负数类似，-1=1，-3=0，以此类推。