咨询PyTorch `torch.split`官方文档中分块尺寸表述的歧义问题

你的理解完全正确。官方文档里“chunks with sizes in dim according to split_size_or_sections”的表述，实际含义就是沿dim方向切分得到的各分块，在该切分维度上的尺寸与split_size_or_sections列表的取值一一对应。

表述歧义说明 #

这段英文容易让人困惑的点是“sizes in dim”的省略写法，它的完整语义是：仅指代各分块在被切分的那个dim维度上的尺寸，并非指分块的所有维度尺寸都和列表值匹配。分块除了切分维度，其余维度的形状和原张量完全一致，不会被改动。

直观示例 #

举个可复现的简单切分场景：

• 输入张量形状为(10, 3)，调用torch.split(x, [2, 5, 3], dim=0)
• 切出的三个分块形状分别为(2,3)、(5,3)、(3,3)，每个分块在dim=0上的长度，严格和列表里的三个元素2、5、3一一对应
• 切分的硬性规则：传入的列表所有元素求和值，必须等于原张量在dim维度的总长度，否则会直接报尺寸不匹配的错误，PyTorch不会自动调整切分长度适配。

内容的提问来源于stack exchange，提问作者a stand-out flamingo