pytorch KeyError: tensor(1)

65 阅读 0 评论 43 点赞

我是靠谱客的博主傲娇奇迹，最近开发中收集的这篇文章主要介绍pytorch KeyError: tensor(1)，觉得挺不错的，现在分享给大家，希望可以做个参考。

pytorch中的tensor不能作为字典的键，输入下面代码会产生错误

import torch
'''pytorch中的tensor和list不能混用'''
a=torch.tensor([1,2,4])
b={1:'a',2:'b',4:'d'}
c=[b[o] for o in a]
print(c)

KeyError: tensor(1)

正确的是要在循环中临时转成numpy

import torch
a=torch.tensor([1,2,4])
b={1:'a',2:'b',4:'d'}
c=[b[o] for o in a.numpy()]
print(c)

['a', 'b', 'd']

此外在GPU上跑的时候情况有点变化，要写成如下，转换numpy要在CPU中完成

c=[b[o] for o in a.cpu().numpy()]

以上就是傲娇奇迹为你收集整理的pytorch KeyError: tensor(1)的全部内容，希望文章能够帮你解决pytorch KeyError: tensor(1)所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。