expected all tensors to be on the same device, but found at least two devices

这个错误提示意思是期望所有的张量都在同一个设备上，但是发现至少有两个设备。这通常是由于在代码中使用了不同的设备（如CPU和GPU）来处理张量，导致张量在不同的设备上。解决方法是将所有的张量都放在同一个设备上，可以使用.to()方法将张量转移到指定的设备上。