Pytorch学习笔记（六）——合并与拆分

学习目标

掌握Pytorch

---

学习内容

合并：cat/stack
拆分：split/chunk（区别在于split按长度进行拆分，chunk按数量进行拆分）

---

1、torch.cat()

a = torch.rand(4,32,8)
a.shape  #torch.Size([4, 32, 8])
b = torch.rand(5,32,8)
b.shape  #torch.Size([5, 32, 8])

torch.cat([a,b],dim = 0).shape  #torch.Size([9, 32, 8])

note: 在使用torch.cat() 时，非合并维度必须保持一致。

a1 = torch.rand(4,3,32,32)
a2 = torch.rand(5,3,32,32)
torch.cat([a1,a2],dim = 0)