RuntimeError: ‘lengths‘ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor

用CPU训练的时候不会报错，用GPU（cuda）训练时报了这个错。
国内博客基本没找到，上谷歌搜索了下。
只需要把 lengths 张量转化为cpu形式即可。在自己得到训练集的方法中，在return中直接返回lengths.cpu()。我的是这样的：

具体原因我还不知道，是与nn.utils.rnn.pack_padded_sequence()这个函数相关，可能是版本/设备的问题。