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()
原文:https://blog.csdn.net/moshiyaofei/article/details/90519430
https://blog.csdn.net/marukoheart/article/details/88091409 python之Numpy的行列维度、axis=0/1/2...便捷记忆方法
https://www.zhihu.com/question/22686450 python中字符串 s[ : -1]是什么意思?
https://blog.csdn.net/appleml/article/details/80143212 pytorch中contiguous()