PyTorch | Tensor、Numpy、Variable之间的转换
生活随笔
收集整理的這篇文章主要介紹了
PyTorch | Tensor、Numpy、Variable之间的转换
小編覺得挺不錯(cuò)的,現(xiàn)在分享給大家,幫大家做個(gè)參考.
其中,Numpy和Variable之間的轉(zhuǎn)換可以通過tensor推導(dǎo)出來。
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之間的相互轉(zhuǎn)換就可以了,能用推導(dǎo)記憶的,就不需要死記硬背。
總結(jié)
以上是生活随笔為你收集整理的PyTorch | Tensor、Numpy、Variable之间的转换的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: pytorch:一维线性回归(二)
- 下一篇: 剖析 | torch.nn.functi