torch.stack()詳解

torch.stack() 跟torch.scatter()類似，scatter()請看我的上一篇博客：https://blog.csdn.net/weixin_43496455/article/details/103870889 貼個torch.stack()官方文檔的截圖 dim代表沿着哪個維度進行堆疊 舉個例子： dim=0時：（dim不寫時，默認爲0） a: 2x3 ; b: 2x3 ;

>>阅读原文<<