用python将PytorchTensor转换并保存到GPU上的.mat图像

Question

我是Pytorch的初学者，一个问题困扰了我好几天。我想将Pytorch tensor格式的图像保存为.mat文件。我看了一下，但似乎没有一种直接的方法可以将Pytoch张量转换为.mat文件。我发现的一个可能的解决方案是将其转换为numpy阵列，但由于我使用的是Nvidia GPU，当我尝试将Pytorch张量转换为numpy阵列时，会出现以下错误：

fake_images[0] = fake_images[0].numpy()

TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first

如何在使用GPU时将Pytorch tensor保存到.mat文件

Answer 1

照吩咐你的去做：

from scipy.io import savemat

with torch.no_grad():
  mdic = {'fake_images': [f_.cpu().numpy() for f_ in fake_images]}
  savemat('fake_images.mat', mdic)

您可以看到有关将pytorch张量转换为numpy数组here的更多信息