2025年4月30日 星期三 4:02am

pytorch中tensor与其他数据结构的转化(int, list, array)

本文记录cpu张量、cuda张量、list和array之间的转换关系。 

  1. import torch
  2. import numpy as np
  3. # int -> tensor -> int
  4. a = torch.Tensor(1)
  5. b = a.item()
  6.  
  7. # list -> tensor(cpu)
  8. l0 = [1, 2, 3]
  9. t = torch.Tensor(l0)
  10.  
  11. # tensor(cpu) -> numpy -> list
  12. a = t.numpy()
  13. l1 = t.numpy().tolist()
  14.  
  15. # list -> numpy
  16. a0 = np.array(l0)
  17.  
  18. # numpy -> tensor(cpu)
  19. t1 = torch.from_numpy(a0)
  20.  
  21. # tensor(cpu) -> tensor(cuda)
  22. tc = t1.cuda()
  23.  
  24. # tensor(cuda) -> list
  25. l2 = tc.cpu().numpy().tolist()
  26.

 

文章知识点与官方知识档案匹配，可进一步学习相关知识
算法技能树首页概览60791 人正在系统学习中
注：本文转载自blog.csdn.net的木盏的文章"https://blog.csdn.net/leviopku/article/details/109318837"。版权归原作者所有，此博客不拥有其著作权，亦不承担相应法律责任。如有侵权，请联系我们删除。
复制链接
相关推荐
发表评论

评论记录：

未查询到任何数据！