在Pytorch 中将Tensor x转换为 numpy数组报错:
...
x=x.numpy()
TypeError:can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
于是按照错误提示,改为
x=x.cpu().numpy()
Tensor 成功转为数组。
在Pytorch 中将Tensor x转换为 numpy数组报错:
...
x=x.numpy()
TypeError:can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
于是按照错误提示,改为
x=x.cpu().numpy()
Tensor 成功转为数组。