首页 » 技术分享 » RuntimeError: index out of range at ..\aten\src\TH/generic/THTensorEvenMoreMath.cpp:193

RuntimeError: index out of range at ..\aten\src\TH/generic/THTensorEvenMoreMath.cpp:193

 

python编程语言pytorch框架的一个读取图像数据集的程序,在运行的时候,其中把图像的标签数字转化为onehot编码的语句报错,该语句如下:

torch.sparse.torch.eye(NUM_CLASSES).index_select(dim=0, index=label)

报错如下:

RuntimeError: index out of range at ..\aten\src\TH/generic/THTensorEvenMoreMath.cpp:193

这因为label= -1造成的数组越界,图像的标签数字最小为0,减去1之后输入报错语句就越界了,

解决办法:label值都加上1就解决了。

转载自原文链接, 如需删除请联系管理员。

原文链接:RuntimeError: index out of range at ..\aten\src\TH/generic/THTensorEvenMoreMath.cpp:193,转载请注明来源!

0