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]

>>阅读原文<<