维度变换之Squeeze vs unsqueeze

常见API

• view / reshape
• squeeze / unsqueeze
• transpose / t / permute
• expand / repeat

Squeeze / unsqueeze

• unsqueeze
  注意：1、a.unsqueeze(i), 其中i值的范围为：[-a.dim() - 1, a.dim() + 1) -> [-5, 5)，左开右闭，所以当i=5时报错。
  2、i值就表示新张量新增的维度，例如a.unsqueeze(0)，就表示新向量的第0维是增加的，新向量torch.Size([1, 4, 1, 28, 28])的第0维度是增加的。
  
  
  unsqueeze(i)方法可以连续使用，但要注意后一个unsqueeze是用在前一个新张量的基础上的。例如unsqueeze(2)是用于b.unsqueeze(1)[形状为torch.Size([32, 1])， 维度为2]上的，而不是用于b[形状为torch.Size([32])， 维度为1]上的。
  

• squeeze