torch.stack()的用法

torch.stack((tensor1, tensor2), dim = x)用法相当于将tensor进行堆叠。

当dim = 0时，将tensor1和tensor2进行堆叠；

dim = 1时，将tensor1[0]和tensor2[0]进行堆叠；

dim = 2时，将tensor1[0][0]和tensor2[0][0]进行堆叠；