pytorch numpy list类型之间的相互转换实例

时间:2021-05-22

如下所示:

import torchfrom torch.autograd import Variableimport numpy as np'''pytorch中Variable与torch.Tensor类型的相互转换''' # 1.torch.Tensor转换成Variablea=torch.randn((5,3))b=Variable(a)print('a',a.type(),a.shape)print('b',type(b),b.shape) # 2.Variable转换成torch.Tensorc=b.data#通过 Variable.data 方法相当于将Variable中的torch.tensor 取出来print('c',c.type(),c.shape) '''torch.tensor与numpy之间的相互转换'''# 3.torch.tensor转换成numpyd=c.numpy()# 4.numpy转换成torch.tensore=torch.from_numpy(d)print('d',type(d))print('e',type(e)) '''numpy和list之间的相互转换 注意这种转换只支持one-dimension array'''# 5.numpy转换成listf1=d.tolist()f2=list(d)# 6.list转换成numpyg=np.asarray(f2)print('f1',type(f1))print('f2',type(f2))print('g',type(g))'''a torch.FloatTensor torch.Size([5, 3])b <class 'torch.Tensor'> torch.Size([5, 3])c torch.FloatTensor torch.Size([5, 3])d <class 'numpy.ndarray'>e <class 'torch.Tensor'>f1 <class 'list'>f2 <class 'list'>g <class 'numpy.ndarray'>'''

以上这篇pytorch numpy list类型之间的相互转换实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章