两个张量在同一个设备上,但我得到了一个错误:预期所有张量都在同一个设备上,但至少找到了两个设备,cuda:0和cpu 我有: def loss_fn(self, pred, truth): truth_flat = torch.reshape(truth, (truth.size(0), ...2024-10-03 已阅读: n次