不經常用PyTorch, 所以也不熟悉裏面的各種常用操作. 簡單修改自己的代碼時最常見到的的問題是忘了各種數據類型之間如何轉換, 記錄常用轉換過程:
同時, 其它的常用方法:
- Tensor的維度信息: *.Size()
- 將一個c*h*w的圖像讀入到張量中, 現在需要把它reshape到四維:
*.unsqueeze(0)
or*.view(1, *(size))
參考:
Pytorch變量類型轉換
https://stackoverflow.com/questions/43328632/pytorch-reshape-tensor-dimension
http://blog.csdn.net/manong_wxd/article/details/78590754