不经常用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