Pytorch chunk的方法做的是对张量进行分块,返回一个张量列表。但如果指定轴的元素个数被chunks除不尽,最后一块的元素个数会少。
torch.chunk(tensor, chunks, dim=0) ->得到一个list的tensors
这个函数的作用是把一个tensor划分到特定数目的块。
| chunks的取值 | 结果 | 
|---|---|
| chunks=n | 代表切分成几个小块 | 
| dim= n | 代表要在哪个维度进行操作 | 

数据横向切割图.png
如果竖着切可以看到如下情况

数据竖着切割图.png
| chunks的取值 | 结果 | 
|---|---|
| chunks=n | 代表切分成几个小块 | 
| dim= n | 代表要在哪个维度进行操作 |