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