RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0

看到这个报错我就想，是不是只要涉及到Tensor的地方，都要带一个 to(device)，此处的device指的是cuda

于是我找到对应的没有采用 to(device) 的但涉及到 tensor 的代码，将以下部分：

actions_v.unsqueeze(-1).type(torch.LongTensor)

改成了：

actions_v.unsqueeze(-1).type(torch.LongTensor).to(device)

重新运行代码，果然不再报错了。

报错的本意就是：希望所有的tensor都在同一个设备上，而不是一会cpu，一会儿gpu