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]进行堆叠;
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]进行堆叠;