个人遇到的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,所以一定记得加()