pytorch KeyError: tensor(1)

77 阅读 0 评论 51 点赞

我是靠谱客的博主傲娇奇迹，这篇文章主要介绍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内容请搜索靠谱客的其他文章。

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