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()