torch.cat很直观,不改变维度数,二维数组和二维数组cat拼接还是二维数组
对两个2维tensor(分别为23,13)进行拼接,拼接完后变为3*3还是2维的tensor
torch.stack,会增加一个维度
两个12维的tensor在第0个维度上stack,则会变为212的tensor;在第1个维度上stack,则会变为12*2的tensor
torch.cat很直观,不改变维度数,二维数组和二维数组cat拼接还是二维数组
对两个2维tensor(分别为23,13)进行拼接,拼接完后变为3*3还是2维的tensor
torch.stack,会增加一个维度
两个12维的tensor在第0个维度上stack,则会变为212的tensor;在第1个维度上stack,则会变为12*2的tensor