pytorch中數據合併與拆分--cat,stack,split,chunk

合併

cat

要進行cat合併的維度可以不同，但是其餘維度必須要相同！

stack

創建一個新的維度，要stack的兩個tensor維度要相同！

用處：將兩個班級的學生和成績合併，生成維度含義：2個表格，每個表格32個學生，每個學生考了8門試！

拆分

split 根據長度拆分

split([2,1],dim=0):在第0維度上，拆分爲兩個部分，第一部分是2個，第二部分數量是1個。

chunck 根據數量拆分

chunk(2,dim=0)在第0維度上，將數據拆分爲2塊!