pytorch擴展一個tensor到多維

在使用神經網絡的過程中，經常會用到把一個tensor擴展一個或多個維度的情況，然後把擴展後的維度用來廣播。

• 擴展一個維度的情況，使用unsqueeze()函數，縮小一個維度使用squeeze()函數。參數是dim，是一個int。也就是說只能擴展或縮小一個維度。

t = torch.tensor([[1,2,3]])
t = t.unsqueeze(0)
#t.shape
#[1, 1, 2]

• 如果想擴展多個維度怎麼辦，答案是用tensor的切片來做

# 把一個[2,3]的tensor擴展到[2,1,1,3]的tensor
>>> a = torch.tensor([[1,2,3],[4,5,6]])
>>> a
tensor([[1, 2, 3],
        [4, 5, 6]])
>>> b = a[:,None,None,:]
>>> b
tensor([[[[1, 2, 3]]],


        [[[4, 5, 6]]]])
>>> b.shape
torch.Size([2, 1, 1, 3])
>>>