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()
原文地址:https://www.cnblogs.com/xiximayou/p/11979401.html