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] 的 [

>>阅读原文<<