pytorch tensor在CPU和GPU之间转换，numpy之间的转换
# input = input.cpu().numpy()
input = input.cpu().detach().numpy() # 有grad
- CPU tensor转GPU tensor：
cpu_imgs.cuda()
- GPU tensor 转CPU tensor：
gpu_imgs.cpu()
- numpy转为CPU tensor：
torch.from_numpy( imgs )
4.CPU tensor转为numpy数据：
cpu_imgs.numpy()
-
注意：GPU tensor不能直接转为numpy数组，必须先转到CPU tensor
-
如果tensor是标量的话，可以直接使用 item() 函数（只能是标量）将值取出来：
print loss_output.item()
参考链接：https://blog.csdn.net/moshiyaofei/article/details/90519430