在将 tensor 格式转换成 numpy 格式时出现报错：
TypeError: can't convert cuda:0 device type tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first.
错误原因
需要先将 tensor 转换到 CPU ，因为 Numpy 是 CPU-only
解决方案
使用 .cpu() 先进行转换
修改前：
predict.numpy()
修改后：
predict.cpu().numpy()
