这篇博客讨论了一个在使用PyTorch时遇到的`IndexError: Dimension out of range`错误。问题源于尝试在一个一维向量上进行按维度切分操作，而该向量的形状为[2]，不包含指定的维度1。解决方案是通过更改`torch.chunk`函数的dim参数，将其设置为0或-1。修正后的代码避免了维度越界，确保了正确地将向量分割成两部分。
Bug：IndexError: Dimension out of range (expected to be in range of [-1, 0], but got 1)
代码：
a=torch.tensor([2,1]) b=torch.chunk(a,2,1)
错误原因：
print(a.shape) torch.Size([2])
a 的 shape 是[2] ，不是[1,2]，是一个一维向量，因此 不存在 dim=1的情况
修改
b=torch.chunk(a,2,0) b=torch.chunk(a,2,-1)
6718
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
