1. 程式人生 > 程式設計 >Pytorch資料拼接與拆分操作實現圖解

Pytorch資料拼接與拆分操作實現圖解

1、cat:拼接

直接合並資料

Pytorch資料拼接與拆分操作實現圖解

2、stack拼接:

與cat不同的是,stack建立了一個新的維度,在拼接的同時,給資料增加了類別。並且stack的所有資料維度必須一致。

Pytorch資料拼接與拆分操作實現圖解

3、split拆分:

通過資料長度進行拆分

Pytorch資料拼接與拆分操作實現圖解

4、chunk拆分:

與split不同的是:chunk是指定拆分的個數,將資料拆分為指定個數。

Pytorch資料拼接與拆分操作實現圖解

以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。