当前位置: 首页 > article >正文

Pytorch, tensor存储机制

tensor在内存的储存,分为两个部分(也就是说一个tensor占用了两个内存位置),一个内存储存了这个tensor的形状size、步长stride、数据的索引等信息,我们把这一部分称之为头信息区(Tensor);另一个内存储的就是真正的数据,我们称为存储区 (Storage)。

类似TCP IP,地址头,及存储空间。C语言的结构体也可以实现这种。

1)storage().data_ptr()

(Pdb) xx= batch_T.storage().data_ptr()
(Pdb) p xx
2506820080832
(Pdb)  batch_T[0,0].storage().data_ptr()
2506820080832
(Pdb)  batch_T[0,1].storage().data_ptr()
2506820080832
(Pdb)  batch_T[0,2].storage().data_ptr()
2506820080832

可见不同元素的头位置是一样的,

2)storage_offset()可以得到偏移

(Pdb)  batch_T[0,0,0].storage_offset()
0
(Pdb)  batch_T[0,2,0].storage_offset()
4
(Pdb)  batch_T[0,4,0].storage_offset()
8
(Pdb)  batch_T[0,4,0].dtype
torch.float64

(Pdb)  batch_T[0,4,0]
tensor(-0.0550, dtype=torch.float64)

这个可以得到偏移,但是实际内存地址还是不知道。每个都会带个类型?


http://www.kler.cn/a/17897.html

相关文章:

  • 多元统计分析-聚类分析的原理与应用
  • 大数据技术之SparkSQL——数据的读取和保存
  • springboot+jsp商务安全邮箱(源码+文档)
  • Python代码学习之给图片添加文字或图片水印
  • UPF learing3:TRANS-11
  • python:可以求解Ax=b的库
  • E. Sergey and Subway(思维 + dp)
  • 入门力扣自学笔记264 C++ (题目编号:2432)
  • 网页和原生程序的交互方案
  • 17组漫画卡通字体推荐给设计师
  • 深入理解Python中的生成器和迭代器
  • ipad有必要用手写笔吗?电容笔和Apple pencil区别
  • 智安网络|网络安全威胁越来越多,教你如何全方面应对
  • PMP|敏捷高分口诀,迅速码住!
  • 单例模式的介绍
  • Yolov1 源码讲解 loss.py
  • 【C++】 类练习---封装链表、人物移动
  • gitlab使用docker简单快速部署
  • 数字座舱带动液晶仪表升级,哪些企业「领跑」前装量产份额
  • 20. 资源的调度——Node 亲和性(Node Affinity)