torch.stack()的使用

废话很少说直接上图函数 如图有a, b, c 三个 3x3的Tensor, spa 若是我想把这三个tensor的最后一个维度的元素相叠加，造成一个新的tensorblog 输入 d=torch.stack( (a,b,c) ,dim = 2)im 就会发现他们在第二个维度像叠加，具体来讲变化以下图所示d3 d[0][0]的位置是由a[0][0]  的 [1]   , b[0][0] 的 [10]

>>阅读原文<<