Fork me on GitHub

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()
posted @ 2019-12-03 20:58  西西嘛呦  阅读(369)  评论(0编辑  收藏  举报