常见错误 RuntimeError: expected type torch.FloatTensor but got torch.cuda.FloatTensor
https://www.jianshu.com/p/0be7a375bdbe
https://blog.csdn.net/qq_38410428/article/details/82973895
计算中有的参数为cuda型，有的参数却是cpu型，就会遇到这样的错误。
解决办法：
该加.cuda()的加上，不该用.cpu()的地方去掉它。
