PyTorch | Tensor、Numpy、Variable之间的转换

其中,Numpy和Variable之间的转换可以通过tensor推导出来。

Numpy->Variable:拆分为两步,

Numpy->Tensor: torch.from_numpy(test)

Tensor->Variable:Variable(test)

组合在一起就是:Variable(torch.from_numpy(test))

同样的道理,Variable->Numpy:拆分为两步,

Variable->Tensor:test.data()

Tensor->Numpy:test.numpy()

组合起来就是:test.data.numpy()

所以只需要记忆:Numpy和Tensor以及Tensor和Variable之间的相互转换就可以了,能用推导记忆的,就不需要死记硬背。

猜你喜欢

转载自blog.csdn.net/songxiaolingbaobao/article/details/113443797