解决方法:将数组下标强转为int
例:
# 原:
length = [id2tags[i.item()] for i in y[0].cpu() if i.item() > 0]
# 改:
length = [id2tags[int(i.item())] for i in y[0].cpu() if i.item() > 0]
例:
# 原:
length = [id2tags[i.item()] for i in y[0].cpu() if i.item() > 0]
# 改:
length = [id2tags[int(i.item())] for i in y[0].cpu() if i.item() > 0]