RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:4 and cuda:5!

这个错误通常是因为在同一个操作中使用了不同的设备（例如CPU和GPU）上的张量。为了解决这个问题，你需要确保所有的张量都在同一个设备上。你可以使用`.to(device)`方法将张量移动到指定的设备上。如果你想将所有的张量都移动到同一个设备上，你可以使用以下代码：

device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
tensor.to(device)

其中，`cuda:0`表示将张量移动到第一个GPU上，如果你有多个GPU，你可以使用`cuda:1`、`cuda:2`等来指定不同的GPU。如果你没有GPU，你可以将`device`设置为`cpu`。