CUDA的tensor张量--->CPU的numpy

个人遇到的bug记录

在cpu上运行tensor张量,会出现如下报错

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

报错里面的内容就已经告诉我们怎么改了,即在数据后面加.cpu()

data=data.cpu().numpy()#data是tensor类型

其中data可以是一维,二维,三维等等

注意cpu后面加(),否则会报错

'builtin_function_or_method' object has no attribute 'numpy'

因为data.cpu后,会将data的类型改为builtin_function_or_method,所以一定记得加()

你可能感兴趣的:(bug,numpy,python,深度学习)