pytorch--基础类型之间的转换

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

在pytorch自己定义张量并进行计算的时候，往往会因为类型不匹配而报错，这里稍微记下pytorch之间的类型转换：

对tensor基础类型进行转换：比如说int()、float()、long()、double()、byte()等，直接.类型即可，例如
float()->int：data.int()
GPU与CPU类型之间的转换：
GPU->CPU：data.cpu()
CPU->GPU：data.cuda()
Variable与Tensor：貌似Variable已经被合并到Tensor中了；
numpy与Tensor之间的转换：
numpy->tensor：data.from_numpy
tensor->numoy：data.numpy()