Python 关于Pytorch'torch.split'文档的混淆

当我查看PyTorch中的功能说明时，我发现作为非英语使用者，我很难阅读：

火炬。分裂（张量、分裂大小或分裂部分，尺寸=0）

[……]

如果

split\u size\u或\u sections

是一个列表，则将拆分

tensor

分为

len（分割大小部分）

块，大小根据 要

拆分\u大小\u或\u节

---

“尺寸在

dim

中”是指“尺寸在

split\u size\u或沿着尺寸在
dim
的部分中”？
别担心-你的英语很好，这一行有点混乱

是的，你说得对。这意味着，如果您传递一个列表，例如
split\u size\u或\u sections=[1,2,4,5]
，它将把张量拆分为
len（[1,2,4,5]）
块（拆分发生在
dim
），每个块的长度分别为
1
、
2
、
4
、
5

这隐含地假设
sum（[1,2,4,5]）
等于
dim
的大小，否则将返回错误