Pytorch chunk

Pytorch chunk的方法做的是对张量进行分块，返回一个张量列表。但如果指定轴的元素个数被chunks除不尽，最后一块的元素个数会少。

torch.chunk(tensor, chunks, dim=0) ->得到一个list的tensors

这个函数的作用是把一个tensor划分到特定数目的块。

chunks的取值	结果
chunks=n	代表切分成几个小块
dim= n	代表要在哪个维度进行操作

数据横向切割图.png

如果竖着切可以看到如下情况

数据竖着切割图.png