关于 Pytorch `torch.split` 文档的困惑

Question

当我在 PyTorch 中查看函数 torch.split 的解释时，我发现作为非英语人士的我很难阅读：

<块引用>

torch.split(tensor, split_size_or_sections, dim=0)

[...]

如果 split_size_or_sections 是一个列表，那么 tensor 将被拆分 根据 len(split_size_or_sections) 大小分成 dim 块 到split_size_or_sections。

“尺寸在 dim 中”是否意味着“尺寸在 split_size_or_sections 中沿尺寸 dim”？

Answer 1

别担心 - 你的英语很好，那句台词有点混乱。

是的，你说得对。这意味着如果你传递一个列表，例如split_size_or_sections=[1,2,4,5] 它将张量拆分为 len([1,2,4,5]) 块（拆分发生在 dim 之间），每个块的长度为 1, 2, { {1}}、4 分别。

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