最后预测部分:predict_word = idx_to_word[predict_label.data[0][0]],报错了:KeyError: tensor(81);改为predict_word = idx_to_word[int(predict_label.data[0][0])] 可行。