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()]