解决方法：将数组下标强转为int
例：
复制代码
1
2
3
4# 原： 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:内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复