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)

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

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值