pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换
今天小编就为大家分享一篇pytorch实现张量tensor,图片,CPU,GPU,数组等的转换,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
1,创建pytorch的Tensor张量:
torch.rand((3,224,224))#创建随机值的三维张量,大小为(3,224,224)torch.Tensor([3,2])#创建张量,[3,2]
2,cpu上的tensor和GPU即pytorch创建的tensor的相互转化
b=a.cpu()#GPU→CPUa=b.cuda()#CPU→GPU
3,tensor和numpy的转化
<pre