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

合併 cat 要進行cat合併的維度可以不同,但是其餘維度必須要相同! stack 創建一個新的維度,要stack的兩個tensor維度要相同! 用處:將兩個班級的學生和成績合併,生成維度含義:2個表格,每個表格32個學生,每個學生考了8門試! 拆分 split 根據長度拆分 split([2,1],dim=0):在第0維度上,拆分爲兩個部分,第一部分是2個,第二部分數量是1個。 chunck 根
相關文章
相關標籤/搜索