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

这个错误提示通常是因为在使用PyTorch时，有些张量被放置在CPU上，有些张量被放置在GPU上，而在某些操作中需要所有的张量都在同一个设备上。为了解决这个问题，你需要将所有的张量都移动到同一个设备上。

你可以使用`.to()`方法将张量移动到指定设备，例如：

import torch

device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

tensor1 = torch.tensor([1, 2, 3]).to(device) # 将张量tensor1移动到指定设备
tensor2 = torch.tensor([4, 5, 6]).to(device) # 将张量tensor2移动到指定设备

# 此时两个张量都在同一个设备上，不会出现Expected all tensors to be on the same device的错误