Pytorch chunk的方法做的是对张量进行分块，返回一个张量列表。但如果指定轴的元素个数被chunks除不尽，最后一块的元素个数会少。
torch.chunk(tensor, chunks, dim=0) ->得到一个list的tensors
这个函数的作用是把一个tensor划分到特定数目的块。

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

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