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

不想你离开。 提交于 2019-12-06 10:54:48

在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()
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!