当试图将存储在GPU(cuda)中的张量值转换为numpy时,假设张量变量为x,然后,使用x.numpy()会引发:TypeError: can’t convert CUDA tensor to numpy,这样的错误。
其实解决方案很简单,通过将数据恢复到cpu然后使用numpy即可,如下所示:
修改为:x.cpu().numpy()
当试图将存储在GPU(cuda)中的张量值转换为numpy时,假设张量变量为x,然后,使用x.numpy()会引发:TypeError: can’t convert CUDA tensor to numpy,这样的错误。
其实解决方案很简单,通过将数据恢复到cpu然后使用numpy即可,如下所示:
修改为:x.cpu().numpy()