Pytorch中variable,tensor和numpy相互转换

熟悉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()
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容