这段代码是合法的,可以正确运行。top5_indices[i][j]是可以访问的。

报错信息可能是因为你尝试访问的索引值不是整数,而是一个数组。你可以尝试将top5_indices[i][j]转换为整数,例如使用int()函数进行强制类型转换,然后再进行访问。

Python PyTorch topk函数使用及访问索引

原文地址: https://www.cveoy.top/t/topic/qfKv 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录