masked_image[i, :, :] *= mask RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
时间: 2023-10-28 07:52:03 浏览: 50
ibm.m.rar_IBM matlab_The First_binary mask_ideal binary mask
这是一个运行时错误,它表示您的代码中存在与设备相关的问题。您的张量必须在同一个设备上。您可以使用 `.to()` 方法将张量尽可能地移动到同一设备上,例如将 CPU 上的所有张量移动到 GPU 上,或者将 GPU 上的所有张量移动到 CPU 上。您还可以检查代码中是否有异步计算或未正确使用 `.to()` 方法。
阅读全文