pytorch 速记 | squeeze,unsqueeze, cat,stack, view 常见用法
在实际使用中老是忘记这些和维度相关操作的API用法,在这里记录一下,作为备忘。unsqueezeunsqueeze指定某一个维度(如0),然后在该维度再插入一个维度,一般用来增加维度,尺寸只能为1。从以下代码可以看出,经过unsuqeeze之后,tensor的尺寸在0的位置多了一个1。In[2]:a=torch.tensor([1,2,3])In[3]:a=a.unsqueeze(0)In[4]: