python stack函數等

torch.stack()函數其實等於是在堆疊數據。
對於一個二維的tensor類型數據來說，距離如下：

import torch
a=torch.tensor([[1,2,3],[4,5,6],[7,8,9]])
a

tensor([[1, 2, 3],
        [4, 5, 6],
        [7, 8, 9]])

這裏的*維變量其實就等於是這個tensor裏的unit數據的維度。

torch.cat(input,dim=0)

默認按行連接張量