Pytorch中variable,tensor和numpy相互转
2020-06-06 本文已影响0人
抄书侠
熟悉python 的同学对numpy模块应该很熟悉,pytorch的数据结构和numpy又极为相似。所以在使用的时候有的时候用numpy很顺手就写出来了,可以逐句变换成pytorch中tensor或variable的操作。但是numpy和tensor也并不完全是一一对应的,所以我们还可以选择把想对数据做的变换先从tensor变成numpy,在numpy下处理了,然后转换到tensor。
1.numpy -> Tensor
a = torch.from_numpy(a)
2.Tensor -> numpy
a = a.numpy()
3.numpy -> Variable
from troch.autograd import Variable
a = Variable(torch.from_numpy(a))
4.Variable -> numpy
a = a.data.numpy()