a = torch.softmax(a, dim=1)
a = a.numpy()在CPU上是正常运行的，然后用GPU的时候就出现了这个报错。
TypeError: can’t convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
numpy不能直接读取CUDA tensor，需要将它转化为 CPU tensor。如果想把CUDA tensor格式的数据改成numpy，需要先将其转换成cpu float-tensor之后再转到numpy格式。
他已经告诉我们修改方法了，要先把 a 修改成 a.cpu()
a = a.cpu().numpy()改成这个样子就好了！
修改过程中，第一次改的时候忘记加括号了，改成了：
a = a.cpu.numpy()会出现新的报错，记得把括号加上！
'builtin_function_or_method' object has no attribute 'numpy'
还有问题可以参考这个文章看看：pytorch:tensor与numpy转换