在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()
原文地址:https://www.cnblogs.com/xiximayou/p/11979401.html
时间: 2024-11-06 09:59:27