Pytorch出现的小问题

Pytorch问题及解决：RuntimeError:‘lengths‘ argument should be a 1D CPU int64 tensor, but got 1D cuda:0 Long tensor*

背景：在运行Pytorch的GPU版本上运行代码时出现以上报错内容

分析：

可以发现以上报错内容提示，参数“lengths”应该使用CPU int64，经过查阅出现以上报错的原因可能是由于Pytorch版本的问题，不能使用GPU版本进行代码的运行，此时，可以改用CPU进行代码尝试

解决方法：

直接在报错指定的地方改成CPU进行运算。即调用 .cpu() 直接添加在报错参数的后面。如下图