我是靠谱客的博主 激动板凳，最近开发中收集的这篇文章主要介绍【pytorch】【Error】KeyError: tensor(2, device=‘cuda:0‘)，觉得挺不错的，现在分享给大家，希望可以做个参考。
概述
解决方法：将数组下标强转为int
例：
# 原：
length = [id2tags[i.item()] for i in y[0].cpu() if i.item() > 0]
# 改：
length = [id2tags[int(i.item())] for i in y[0].cpu() if i.item() > 0]最后
以上就是激动板凳为你收集整理的【pytorch】【Error】KeyError: tensor(2, device=‘cuda:0‘)的全部内容，希望文章能够帮你解决【pytorch】【Error】KeyError: tensor(2, device=‘cuda:0‘)所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复