【pytorch】合并与分割

拆分

Cat

stack

Split 按长度进行拆分

Chunk按数量进行拆分

Cat

我们想要合并dim=0如何合并呢

In:torch.cat([a,b],dim=0).shape
###意思是将dim=0进行合并，此时必须保证的是其他的dim都是一样的数

Out:torch.Size([9,32,8])

Cat使用的条件：必须是维数相同，除合并 的dim外都要是相同的数

巩固：

stack

巩固：

Split 通过长度来分开

Chunk 通过数来分开