torch.stack()的使用

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

>>阅读原文<<