Pytorch torch.cat與torch.stack的區別

torch.cat()函數可以將多個張量拼接成一個張量。torch.cat()有兩個參數，第一個是要拼接的張量的列表或是元組；第二個參數是拼接的維度。 torch.cat()的示例如下圖1所示                         圖1 torch.cat() torch.stack()函數同樣有張量列表和維度兩個參數。stack與cat的區別在於，torch.stack()函數要求輸入

>>阅读原文<<