pytorch的tensor与numpy数组共享值

本文纠正了网络上关于Tensor与Numpy数组共享内存的误解,详细解析两者仅部分共享内存,即共享值,且在特定条件下成立。通过代码示例解释了Cpython环境下,Tensor对象与Numpy对象内部指针机制,揭示共享内存的真实含义。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

网上的很多博客说tensor与numpy数组共享内存,这是错误的
先给结论:tensor与numpy数组共享部分内存,说共享值更严谨,且有条件
看代码:
a = torch.ones(2,2)
b = a.numpy()
print(id(a))
print(id(b))
输出:
3030786996336
3030755215040
解析:在Cpython里面,a,b是C指针,torch.ones(2,2)构造生成一个tensor实例,放入一片内存区域,指针a指向这片内存区域,构造生成的tensor对象内部又有多个指针,指向不同的内存区域,其中有一个指向数组的指针tv_ptr,a.numpy()构造生成一个numpy对象,且这个numpy对象内部也有多个指针,其中也有一个指向数组的指针nv_ptr,仅这两个指针指向同一内存区域,并非所有指针都共享内存,b指针指向构造生成的numpy对象

共享值的条件:
tensor对象的操作必须是在原内存区域上的操作,numpy对象的操作也是在原对象内存区域上的操作,2者如何在原内存区域上操作而不是指向新的内存区域,看我上两篇博客

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值