torch.stack()

torch.stack() 含義:在指定的維度對元素進行疊加 例子: 使用torch生成如下的三個tensor: a, b, c 在指定的維度 0 對a, b, c進行疊加: 在指定的維度 1 對a, b, c進行疊加: 在指定的維度 2 對a, b, c進行疊加: 注意: 對於維度 0, 1, 2應該從外往裏面看進去. 在指定的維度對tensor進行疊加時, 應該把該維度對應的個體看成是一個塊,
相關文章
相關標籤/搜索