pytorch 速记 | squeeze，unsqueeze, cat，stack, view 常见用法

在实际使用中老是忘记这些和维度相关操作的API用法，在这里记录一下，作为备忘。

unsqueeze

unsqueeze指定某一个维度（如0），然后在该维度再插入一个维度，一般用来增加维度，尺寸只能为1。
从以下代码可以看出，经过unsuqeeze之后，tensor的尺寸在0的位置多了一个1。

In [2]: a = torch.tensor([1,2,3])                                               

In [3]: a = a.unsqueeze(0)                                                      

In [4]: a                                                                       
Out[4]: tensor([[1, 2, 3]])

In [5]: a.size()                                                                
Out[5]: torch.Size([1, 3])