学习目标
掌握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)