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

标签：基础   ble   cpu   tensor   直接   variable   cuda   span   计算   

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

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

标签：基础   ble   cpu   tensor   直接   variable   cuda   span   计算   

原文地址：https://www.cnblogs.com/xiximayou/p/11979401.html