TypeError: can’t convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor 在报错的变量后面加 .cuda().data.cpu().numpy() 即可解决