Pytorch中变量在tensor与numpy之间转换，tensor在CPU和GPU之间转换

1.CPU tensor转GPU tensor：

cpu_imgs.cuda()

2. GPU tensor 转CPU tensor：

gpu_imgs.cpu()

3. numpy转为CPU tensor：

torch.from_numpy( imgs )

4.CPU tensor转为numpy数据：

cpu_imgs.numpy()

5.GPU tensor转为numpy数据：

gpu_imgs.cpu().numpy()