在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()