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 → CPU
a ``=` `b.cuda() ``#CPU → GPU
3, tens
原创
2021-04-04 21:17:32 ·
473 阅读 ·
0 评论