© 1999-2048 dssz.net 粤ICP备11031372号
[其它] 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, tensor和numpy的转化 b = a.numpy() # tensor转化为 numpy<weixin_38506182> 上传 | 大小:36kb