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

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

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

 

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

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

 

3.将numpy转换为Variableclass

sub_va = Variable(torch.from_numpy(sub_img))

 

4.将Variable张量转化为numpynumpy

sub_np2 = sub_va.data.numpy()
相关文章
相关标签/搜索