Pytorch學習筆記【5】---tensor的拼接和拆分

Pytorch學習筆記【5】—tensor的拼接和拆分 Pytorch筆記目錄:點擊進入 python 文章目錄 Pytorch學習筆記【5】---tensor的拼接和拆分 1.cat 2.stack 3.根據長度來分割 4.經過通道數來分割 1.cat 把兩個向量縱向拼接起來,並不會增長新的維度web # cat a = torch.rand(4,32,8) b = torch.rand(5,3
相關文章
相關標籤/搜索