最常用的3种Pytorch tensor的维度变化方法
最近一周一直在搭建新的模型,学到了很多新的tensor维度变化操作,记录一下:a = torch.tensor([1,2,3])1、None增加一维:b = a[None, :] # 改成两维(一个逗号),增加到第一维>>> b == tensor([[1, 2, 3]])c = a[:, None] # 改成两维(一个逗号),增加到第二维>>> c == tensor([[1],[2],[3]])d = a