torch.split与torch.chunk的用法及区别

torch.split()
参考链接https://blog.csdn.net/foneone/article/details/103875250

torch.split()作用将tensor按照dim分成块结构。

参数：
tesnor：input，待分输入
split_size_or_sections：需要切分的大小(int or list )
dim：切分维度
output：切分后块结构 <class ‘tuple’>

当split_size_or_sections为int时，tenor结构和split_size_or_sections，正好匹配，那么ouput就是大小相同的块结构。dim上数据的长度不能被split_size_or_sections整除，余下的部分会少一些

当split_size_or_sections 为list时，那么tensor结构会一共切分成len(list)这么多的小块，每个小块中的大小按照list中的大小决定，其中list中的数字总和应等于该维度的大小，否则会报错（注意这里与split_size_or_sections为int时的情况不同）。