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. note：GPU tensor不能直接转为numpy数组，必须先转到CPU tensor。
6. 如果tensor是标量的话，可以直接使用 item() 函数（只能是标量）将值取出来：
print loss_output.item()
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
