每天一个芝士点

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()
上一篇下一篇

猜你喜欢

热点阅读