首页 > 其他 > 详细

Pytorch中的variable, tensor与numpy相互转化的方法

时间:2019-04-03 14:59:23      阅读:277      评论:0      收藏:0      [点我收藏+]

1.将numpy矩阵转换为Tensor张量

sub_ts = torch.from_numpy(sub_img) #sub_img为numpy类型

 

2.将Tensor张量转化为numpy矩阵

sub_np1 = sub_ts.numpy() #sub_ts为tensor张量

 

3.将numpy转换为Variable

sub_va = Variable(torch.from_numpy(sub_img))

 

4.将Variable张量转化为numpy

sub_np2 = sub_va.data.numpy()

 

Pytorch中的variable, tensor与numpy相互转化的方法

原文:https://www.cnblogs.com/Archer-Fang/p/10648754.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!