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

你可能感兴趣的:(python)