错误原因:需要int型,但给的数据给的顺序错了
1.
def generation(pred, vocabulary, sent_len=None):
A = generation(data, vocabulary, sent_len)
2. 出错代码 cur_len = sent_len[idx] cur_real_pred = cur_pred[:cur_len]
代码更改
cur_len = sent_len[idx]
cur_real_pred = cur_pred[:cur_len.item()]