pytorch numpy list类型之间的相互转换实例
如下所示：
importtorch
fromtorch.autogradimportVariable
importnumpyasnp
'''
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.Tensor
c=b.data#通过Variable.data方法相当于将Variable中的torch.tensor取出来
print('c',c.type(),c.shape)
'''
torch.tensor与numpy之间的相互转换
'''
#3.torch.tensor转换成numpy
d=c.numpy()
#4.numpy转换成torch.tensor
e=torch.from_numpy(d)
print('d',type(d))
print('e',type(e))
'''
numpy和list之间的相互转换注意这种转换只支持one-dimensionarray
'''
#5.numpy转换成list
f1=d.tolist()
f2=list(d)
#6.list转换成numpy
g=np.asarray(f2)
print('f1',type(f1))
print('f2',type(f2))
print('g',type(g))
'''
atorch.FloatTensortorch.Size([5,3])
btorch.Size([5,3])
ctorch.FloatTensortorch.Size([5,3])
d
e
f1
f2
g
'''
以上这篇pytorchnumpylist类型之间的相互转换实例就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持毛票票。