torch.stack()解析

torch.stack()是將原來的幾個tensor按照一定方式進行堆疊，然後在按照堆疊後的維度進行切分。

有a，b，c三個tensor.

dim=0

dim=1

dim=2

參考：torch.stack(), torch.cat()用法詳解