torch.stack解析以及與torch.cat的關係

torch.stack這個函數看上去輸出的結果很奇怪，但是其實解析之後很簡單。 舉個例子： 然後使用torch.stack函數： 一眼看上去，好像有點懵逼，不急，我們慢慢解析。 這個函數是這樣的，他的dim參數表明他會在dim新建一個維度，然後在這個維度上進行concat。 因此，我們可以這麼理解，假如取dim=1，那麼我們可以將a，b同時在dim=1新增一個維度，即： 然後再在dim=1的維度上

>>阅读原文<<